From 81f3b6f56d3511c7a4dda0e01759e81321c4019f Mon Sep 17 00:00:00 2001 From: Guilherme Aldeia Date: Mon, 19 Feb 2024 09:14:59 -0300 Subject: [PATCH] Removed NoneType from code NoneType is no longer directly importable in newer versions of python --- src/brush/estimator.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/brush/estimator.py b/src/brush/estimator.py index d076dd2c..03553645 100644 --- a/src/brush/estimator.py +++ b/src/brush/estimator.py @@ -12,7 +12,6 @@ # import deap as dp from deap import algorithms, base, creator, tools # from tqdm import tqdm -from types import NoneType from sklearn.metrics import average_precision_score from sklearn.preprocessing import MinMaxScaler import _brush @@ -404,7 +403,7 @@ def _make_data(self, X, y=None, feature_names=[], validation_size=0.0): assert isinstance(X, np.ndarray) - if isinstance(y, NoneType): + if isinstance(y, None): return _brush.Dataset(X=X, feature_names=feature_names, validation_size=validation_size) @@ -605,4 +604,4 @@ def _make_individual(self): # def transform(self, X): # """Transform X using the best estimator in the archive. """ -# return self.predict(X) \ No newline at end of file +# return self.predict(X)