downgrade river

kulbachcedric · Oct 18, 2022 · bf590d2 · bf590d2
1 parent 18bd7ad
commit bf590d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/EvOAutoML/base/evolution.py b/EvOAutoML/base/evolution.py
@@ -6,7 +6,6 @@
 from river import base, compose, metrics, preprocessing, tree
 from river.base import Estimator
 from river.drift import ADWIN
-from river.metrics import ClassificationMetric
 from sklearn.model_selection import ParameterGrid, ParameterSampler
 
 
@@ -83,7 +82,7 @@ def reset(self):
         self._i = 0
         return self
 
-    def _mutate_estimator(self, estimator) -> (base.Classifier, ClassificationMetric):
+    def _mutate_estimator(self, estimator) -> (base.Classifier):
         child_estimator = copy.deepcopy(estimator)
         key_to_change = self._rng.choice(list(self.param_grid.keys()))
         value_to_change = self.param_grid[key_to_change][self._rng.choice(range(len(self.param_grid[key_to_change])))]

diff --git a/setup.py b/setup.py
@@ -17,7 +17,7 @@
     "scikit-learn==1.0.2",
     "scikit-surprise==1.1.1",
     "pandas~=1.3.2",
-    "river~=0.13.0",
+    "river~=0.10.1",
     "tqdm~=4.61.2",
     "pytest==7.0.1",
 ]