fixed linters

MobileTeleSystems · Dec 7, 2024 · 49aada9 · 49aada9
1 parent 1cd5d54
commit 49aada9
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/rectools/models/base.py b/rectools/models/base.py
@@ -96,7 +96,7 @@ class ModelBase(tp.Generic[ModelConfig_T]):
     recommends_for_warm: tp.ClassVar[bool] = False
     recommends_for_cold: tp.ClassVar[bool] = False
 
-    config_class: tp.ClassVar[tp.Type[ModelConfig_T]]
+    config_class: tp.Type[ModelConfig_T]  # ClassVar cannot contain type variables
 
     def __init__(self, *args: tp.Any, verbose: int = 0, **kwargs: tp.Any) -> None:
         self.is_fitted = False

diff --git a/rectools/models/implicit_als.py b/rectools/models/implicit_als.py
@@ -137,7 +137,7 @@ def __init__(self, model: AnyAlternatingLeastSquares, verbose: int = 0, fit_feat
         self.fit_features_together = fit_features_together
         self.use_gpu = isinstance(model, GPUAlternatingLeastSquares)
         if not self.use_gpu:
-            self.n_threads = model.num_threads
+            self.n_threads = model.num_threads  # type: ignore  # TODO: remove when recommend n_threads implemented
 
     @classmethod
     def _make_config(

diff --git a/rectools/models/lightfm.py b/rectools/models/lightfm.py
@@ -130,7 +130,7 @@ def __init__(
         self.model: LightFM
         self._model = model
         self.n_epochs = epochs
-        self.n_threads = num_threads
+        self.n_threads = num_threads  # type: ignore  # TODO: remove when recommend n_threads implemented
 
     def _get_config(self) -> LightFMWrapperModelConfig:
         inner_model = self._model