fixed coverage

MobileTeleSystems · Oct 4, 2023 · cec82e0 · cec82e0
1 parent 2a8d7af
commit cec82e0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/rectools/models/implicit_als.py b/rectools/models/implicit_als.py
@@ -82,7 +82,7 @@ def _fit(self, dataset: Dataset) -> None:  # type: ignore
             self.verbose,
         )
 
-        if self.use_gpu:
+        if self.use_gpu:  # pragma: no cover
             user_factors = implicit.gpu.Matrix(user_factors)
             item_factors = implicit.gpu.Matrix(item_factors)
 
@@ -124,7 +124,7 @@ def get_users_vectors(model: AnyAlternatingLeastSquares) -> np.ndarray:
     np.ndarray
        User vectors
     """
-    if isinstance(model, GPUAlternatingLeastSquares):
+    if isinstance(model, GPUAlternatingLeastSquares):  # pragma: no cover
         return model.user_factors.to_numpy()
     return model.user_factors
 
@@ -143,7 +143,7 @@ def get_items_vectors(model: AnyAlternatingLeastSquares) -> np.ndarray:
     np.ndarray
         Item vectors
     """
-    if isinstance(model, GPUAlternatingLeastSquares):
+    if isinstance(model, GPUAlternatingLeastSquares):  # pragma: no cover
         return model.item_factors.to_numpy()
     return model.item_factors