Merge branch 'main' into vinamarora8-pro-159

neuro-galaxy · Nov 16, 2024 · c4fd3fb · c4fd3fb
2 parents 57290dd + 69865a5
commit c4fd3fb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/examples/poyo_plus/train.py b/examples/poyo_plus/train.py
@@ -15,7 +15,8 @@
 from torch_brain.nn import compute_loss_or_metric
 from torch_brain.registry import MODALITIY_REGISTRY
 from torch_brain.utils import callbacks as tbrain_callbacks
-from torch_brain.utils import seed_everything, DataModule
+from torch_brain.utils import seed_everything
+from torch_brain.utils.datamodules import DataModule
 from torch_brain.utils.stitcher import StitchEvaluator
 
 # higher speed on machines with tensor cores

diff --git a/torch_brain/utils/__init__.py b/torch_brain/utils/__init__.py
@@ -2,5 +2,3 @@
 from .sinusoidal_encoding import get_sinusoidal_encoding
 
 from .seed_everything import seed_everything
-
-from .datamodules import DataModule
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2,5 +2,3 @@
		from .sinusoidal_encoding import get_sinusoidal_encoding

		from .seed_everything import seed_everything

		from .datamodules import DataModule