fix lints in embeddings

sb-ai-lab · Oct 20, 2023 · c573885 · c573885
1 parent 2f97195
commit c573885
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sim4rec/modules/embeddings.py b/sim4rec/modules/embeddings.py
@@ -130,7 +130,7 @@ def __init__(
         self._num_loader_workers = num_loader_workers
         self._max_iter = max_iter
 
-    # pylint: disable=too-many-locals, not-callable, not-an-iterable
+    # pylint: disable=too-many-locals, not-callable
     def _fit(
         self,
         dataset : DataFrame
@@ -140,7 +140,7 @@ def _fit(
         device_name = self.getDevice()
         seed = self.getSeed()
         device = torch.device(self.getDevice())
-
+        # pylint: disable=not-an-iterable
         X = dataset.select(*inputCols).toPandas().values
 
         torch.manual_seed(torch.seed() if seed is None else seed)