diff --git a/python/interpret-core/interpret/glassbox/ebm/test/test_ebm.py b/python/interpret-core/interpret/glassbox/ebm/test/test_ebm.py index 77f251a7d..364847c63 100644 --- a/python/interpret-core/interpret/glassbox/ebm/test/test_ebm.py +++ b/python/interpret-core/interpret/glassbox/ebm/test/test_ebm.py @@ -46,8 +46,8 @@ def test_unknown_multiclass_category(): X_train['cat_feature'] = [np.random.choice(['a', 'b', 'c']) for x in range(X_train.shape[0])] X_test['cat_feature'] = ['d' for x in range(X_test.shape[0])] # Unknown category in test set - X_train['cat_feature'][1] = np.nan - X_test['cat_feature'][1] = np.nan + # X_train['cat_feature'][1] = np.nan + # X_test['cat_feature'][1] = np.nan clf = ExplainableBoostingClassifier() clf.fit(X_train, y_train)