Update estimator.py

Fixed check for None type
cavalab · Feb 19, 2024 · 8de40e1 · 8de40e1
1 parent 81f3b6f
commit 8de40e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brush/estimator.py b/src/brush/estimator.py
@@ -403,7 +403,7 @@ def _make_data(self, X, y=None, feature_names=[], validation_size=0.0):
 
         assert isinstance(X, np.ndarray)
 
-        if isinstance(y, None):
+        if y is None:
             return _brush.Dataset(X=X,
                     feature_names=feature_names, validation_size=validation_size)