Update test_ebm.py

Co-Authored-By: S Jenkins <[email protected]> Co-Authored-By: Harsha-Nori <[email protected]>
interpretml · Jan 13, 2021 · edb7418 · edb7418
1 parent 3eb19f3
commit edb7418
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)