Fix new MyPy 11 errors

online-ml · Aug 29, 2024 · cec4034 · cec4034
1 parent b451bcf
commit cec4034
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/river/compat/sklearn_to_river.py b/river/compat/sklearn_to_river.py
@@ -33,7 +33,7 @@ def convert_sklearn_to_river(estimator: sklearn_base.BaseEstimator, classes: lis
         (sklearn_base.RegressorMixin, SKL2RiverRegressor),
         (
             sklearn_base.ClassifierMixin,
-            functools.partial(SKL2RiverClassifier, classes=classes),
+            functools.partial(SKL2RiverClassifier, classes=classes),  # type:ignore[arg-type]
         ),
     ]
 

diff --git a/river/feature_extraction/vectorize.py b/river/feature_extraction/vectorize.py
@@ -203,7 +203,7 @@ def __init__(
         # Stop word removal
         if self.stop_words:
             self.processing_steps.append(
-                functools.partial(remove_stop_words, stop_words=stop_words)
+                functools.partial(remove_stop_words, stop_words=self.stop_words)
             )
 
         # n-grams