Fix pytests

online-ml · Jun 4, 2023 · c3da229 · c3da229
1 parent 46e6296
commit c3da229
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/deep_river/base.py b/deep_river/base.py
@@ -171,7 +171,7 @@ def clone(self, new_params: dict = {}, include_attributes=False):
         new_params = new_params or {}
         new_params.update(self.kwargs)
         new_params.update(self._get_params())
-        new_params.update({'module': self.module_cls})
+        new_params.update({"module": self.module_cls})
 
         clone = self.__class__(**new_params)
         if include_attributes:

diff --git a/deep_river/classification/zoo.py b/deep_river/classification/zoo.py
@@ -89,6 +89,8 @@ def __init__(
         seed: int = 42,
         **kwargs,
     ):
+        if "module" in kwargs:
+            del kwargs["module"]
         super().__init__(
             module=LogisticRegression.LRModule,
             loss_fn=loss_fn,
@@ -217,6 +219,8 @@ def __init__(
     ):
         self.n_width = n_width
         self.n_layers = n_layers
+        if "module" in kwargs:
+            del kwargs["module"]
         super().__init__(
             module=MultiLayerPerceptron.MLPModule,
             loss_fn=loss_fn,

diff --git a/deep_river/regression/zoo.py b/deep_river/regression/zoo.py
@@ -9,9 +9,6 @@ class LinearRegression(Regressor):
     """
     This class implements a linear regression model in PyTorch.
 
-    Parameters
-    ----------
-
     """
 
     class LRModule(nn.Module):
@@ -32,6 +29,8 @@ def __init__(
         seed: int = 42,
         **kwargs,
     ):
+        if "module" in kwargs:
+            del kwargs["module"]
         super().__init__(
             module=LinearRegression.LRModule,
             loss_fn=loss_fn,
@@ -130,6 +129,8 @@ def __init__(
     ):
         self.n_width = n_width
         self.n_layers = n_layers
+        if "module" in kwargs:
+            del kwargs["module"]
         super().__init__(
             module=MultiLayerPerceptron.MLPModule,
             loss_fn=loss_fn,