type-check fix

luxonis · Nov 15, 2024 · c0bdbfc · c0bdbfc
1 parent 6e0c888
commit c0bdbfc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/luxonis_train/strategies/__init__.py b/luxonis_train/strategies/__init__.py
@@ -1,5 +1,7 @@
+from .base_strategy import BaseTrainingStrategy
 from .triple_lr_sgd import TripleLRScheduler
 
 __all__ = [
     "TripleLRScheduler",
+    "BaseTrainingStrategy",
 ]