Update test case

DataCanvasIO · Jun 30, 2023 · ca43cbd · ca43cbd
1 parent 1065ac1
commit ca43cbd
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/hypernets/tests/model/test_objectives.py b/hypernets/tests/model/test_objectives.py
@@ -109,6 +109,10 @@ def predict_proba(self, X, **kwargs):
  def predict(self, X, **kwargs):
  return self.proba[:, 1] > 0.5
 
+ @property
+ def _estimator_type(self):
+ return 'classifier'
+
 
 class TestPredictionObjective: