Lint

baal-org · Jun 6, 2024 · 02b52cb · 02b52cb
1 parent 466fe61
commit 02b52cb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/baal/active/heuristics/heuristics_gpu.py b/baal/active/heuristics/heuristics_gpu.py
@@ -106,7 +106,9 @@ def predict_on_dataset(
         verbose=True,
     ):
         return (
-            super().predict_on_dataset(dataset, iterations, half, verbose).reshape([-1])  # type: ignore
+            super()
+            .predict_on_dataset(dataset, iterations, half, verbose)
+            .reshape([-1])  # type: ignore
         )
 
     def predict_on_batch(self, data, iterations=1):

diff --git a/baal/experiments/base.py b/baal/experiments/base.py
@@ -20,8 +20,8 @@
 
     TRANSFORMERS_AVAILABLE = True
 except ImportError:
-    BaalTransformersTrainer = Any  # type: ignore
-    TransformersAdapter = Any  # type: ignore
+    BaalTransformersTrainer = None  # type: ignore
+    TransformersAdapter = None  # type: ignore
     TRANSFORMERS_AVAILABLE = False
 
 log = structlog.get_logger(__name__)