From cd3eca288ec88221a6c9cdda3c049814e1aa9af7 Mon Sep 17 00:00:00 2001 From: Vinam Arora Date: Wed, 13 Nov 2024 18:01:46 -0500 Subject: [PATCH 1/2] Fix circular import in torch_brain.models (fix #26) (#27) --- examples/poyo/train.py | 3 ++- torch_brain/utils/__init__.py | 2 -- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/examples/poyo/train.py b/examples/poyo/train.py index 143569c..687de61 100644 --- a/examples/poyo/train.py +++ b/examples/poyo/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 index 67940ee..83e705d 100644 --- 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 From 69865a56d77ca9bff5c54b60cda2d8b86d8e4ba1 Mon Sep 17 00:00:00 2001 From: Vinam Arora Date: Fri, 15 Nov 2024 05:08:24 -0500 Subject: [PATCH 2/2] Rename examples/poyo to examples/poyo_plus (#28) --- examples/{poyo => poyo_plus}/README.md | 0 examples/{poyo => poyo_plus}/configs/base.yaml | 0 examples/{poyo => poyo_plus}/configs/train_mc_maze_small.yaml | 0 examples/{poyo => poyo_plus}/configs/train_poyo_mp.yaml | 0 examples/{poyo => poyo_plus}/train.py | 0 5 files changed, 0 insertions(+), 0 deletions(-) rename examples/{poyo => poyo_plus}/README.md (100%) rename examples/{poyo => poyo_plus}/configs/base.yaml (100%) rename examples/{poyo => poyo_plus}/configs/train_mc_maze_small.yaml (100%) rename examples/{poyo => poyo_plus}/configs/train_poyo_mp.yaml (100%) rename examples/{poyo => poyo_plus}/train.py (100%) diff --git a/examples/poyo/README.md b/examples/poyo_plus/README.md similarity index 100% rename from examples/poyo/README.md rename to examples/poyo_plus/README.md diff --git a/examples/poyo/configs/base.yaml b/examples/poyo_plus/configs/base.yaml similarity index 100% rename from examples/poyo/configs/base.yaml rename to examples/poyo_plus/configs/base.yaml diff --git a/examples/poyo/configs/train_mc_maze_small.yaml b/examples/poyo_plus/configs/train_mc_maze_small.yaml similarity index 100% rename from examples/poyo/configs/train_mc_maze_small.yaml rename to examples/poyo_plus/configs/train_mc_maze_small.yaml diff --git a/examples/poyo/configs/train_poyo_mp.yaml b/examples/poyo_plus/configs/train_poyo_mp.yaml similarity index 100% rename from examples/poyo/configs/train_poyo_mp.yaml rename to examples/poyo_plus/configs/train_poyo_mp.yaml diff --git a/examples/poyo/train.py b/examples/poyo_plus/train.py similarity index 100% rename from examples/poyo/train.py rename to examples/poyo_plus/train.py