fix: type assignment exception

Author: oesteban

src/nifreeze/estimator.py

@@ -99,7 +99,7 @@ def run(self, dataset: DatasetT, **kwargs) -> Self:
         if self._prev is not None:
             result = self._prev.run(dataset, **kwargs)
             if isinstance(self._prev, Filter):
-                dataset = result
+                dataset = result  # type: ignore[assignment]
 
         n_jobs = kwargs.get("n_jobs", None)
 

src/nifreeze/model/base.py

@@ -84,7 +84,7 @@ class BaseModel:
 
     """
 
-    __slots__ = ("_dataset", )
+    __slots__ = ("_dataset",)
 
     def __init__(self, dataset, **kwargs):
         """Base initialization."""
@@ -103,7 +103,7 @@ def fit_predict(self, index, **kwargs) -> np.ndarray:
 class TrivialModel(BaseModel):
     """A trivial model that returns a given map always."""
 
-    __slots__ = ("_predicted", )
+    __slots__ = ("_predicted",)
 
     def __init__(self, dataset, predicted=None, **kwargs):
         """Implement object initialization."""
@@ -129,7 +129,7 @@ def fit_predict(self, *_, **kwargs):
 class ExpectationModel(BaseModel):
     """A trivial model that returns an expectation map (for example, average)."""
 
-    __slots__ = ("_stat", )
+    __slots__ = ("_stat",)
 
     def __init__(self, dataset, stat="median", **kwargs):
         """Initialize a new model."""