fix: mypy errors

CLARIN-PL · Nov 15, 2022 · 3073be2 · 3073be2
1 parent 7413318
commit 3073be2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/embeddings/pipeline/hps_pipeline.py b/embeddings/pipeline/hps_pipeline.py
@@ -39,7 +39,7 @@ class TorchGarbageCollectorOptunaCallback(OptunaCallback):
     def _cleanup() -> None:
         gc.collect()
         if torch.cuda.is_available():
-            torch.cuda.empty_cache()  # type: ignore
+            torch.cuda.empty_cache()
 
     def __call__(self, study: Study, trial: optuna.trial.FrozenTrial) -> None:
         TorchGarbageCollectorOptunaCallback._cleanup()

diff --git a/embeddings/task/lightning_task/lightning_task.py b/embeddings/task/lightning_task/lightning_task.py
@@ -93,7 +93,7 @@ def fit(
             self.trainer.fit(self.model, data)
         except Exception as e:
             del self.trainer
-            torch.cuda.empty_cache()  # type: ignore
+            torch.cuda.empty_cache()
             raise e
 
     @abc.abstractmethod