From 15ccc53d278633fe880516d16aca5aaaf313df04 Mon Sep 17 00:00:00 2001 From: kaylode Date: Sat, 4 Nov 2023 07:13:53 +0000 Subject: [PATCH] :bug: Remove reference between base and ml to fix error --- theseus/base/callbacks/__init__.py | 2 +- theseus/ml/trainer/ml_trainer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/theseus/base/callbacks/__init__.py b/theseus/base/callbacks/__init__.py index 3d1da88..1de832e 100644 --- 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 index 2c64b93..00bbe0a 100644 --- 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")