🐛 Remove reference between base and ml to fix error

kaylode · Nov 4, 2023 · 15ccc53 · 15ccc53
1 parent c204b4d
commit 15ccc53
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/theseus/base/callbacks/__init__.py b/theseus/base/callbacks/__init__.py
@@ -5,7 +5,7 @@
     RichModelSummary,
 )
 
-from theseus.ml.callbacks.base_callbacks import Callbacks, CallbacksList
+# from theseus.ml.callbacks.base_callbacks import Callbacks, CallbacksList
 from theseus.registry import Registry
 
 from .checkpoint_callback import TorchCheckpointCallback

diff --git a/theseus/ml/trainer/ml_trainer.py b/theseus/ml/trainer/ml_trainer.py
@@ -1,5 +1,5 @@
-from theseus.base.callbacks import CallbacksList
 from theseus.base.utilities.loggers.observer import LoggerObserver
+from theseus.ml.callbacks import CallbacksList
 
 LOGGER = LoggerObserver.getLogger("main")