Move tui module to bootstrap

DubiousCactus · Jul 27, 2024 · 19e08f6 · 19e08f6
1 parent 4aea72d
commit 19e08f6
Show file tree

Hide file tree

Showing 8 changed files with 9 additions and 9 deletions.
diff --git a/bootstrap/launch_experiment.py b/bootstrap/launch_experiment.py
@@ -34,10 +34,10 @@
     make_training_loss,
     parallelize_model,
 )
+from bootstrap.tui.training_ui import TrainingUI
 from conf import project as project_conf
 from src.base_tester import BaseTester
 from src.base_trainer import BaseTrainer
-from tui.training_ui import TrainingUI
 from utils import load_model_ckpt, to_cuda_
 
 console = Console()

diff --git a/tui/__init__.py → bootstrap/tui/__init__.py b/tui/__init__.py → bootstrap/tui/__init__.py
diff --git a/tui/styles/training_ui.css → bootstrap/tui/styles/training_ui.css b/tui/styles/training_ui.css → bootstrap/tui/styles/training_ui.css
diff --git a/tui/training_ui.py → bootstrap/tui/training_ui.py b/tui/training_ui.py → bootstrap/tui/training_ui.py
@@ -28,9 +28,9 @@
 from torchvision.datasets import MNIST
 from torchvision.transforms.functional import to_tensor
 
-from tui import Plot_BestModel, Task
-from tui.widgets.plotting import PlotterWidget
-from tui.widgets.progress import DatasetProgressBar
+from bootstrap.tui import Plot_BestModel, Task
+from bootstrap.tui.widgets.plotting import PlotterWidget
+from bootstrap.tui.widgets.progress import DatasetProgressBar
 
 
 class TrainingUI(App):

diff --git a/tui/widgets/plotting.py → bootstrap/tui/widgets/plotting.py b/tui/widgets/plotting.py → bootstrap/tui/widgets/plotting.py
@@ -5,7 +5,7 @@
 from textual.reactive import var
 from textual_plotext import PlotextPlot
 
-from tui import Plot_BestModel
+from bootstrap.tui import Plot_BestModel
 
 
 class PlotterWidget(PlotextPlot):

diff --git a/tui/widgets/progress.py → bootstrap/tui/widgets/progress.py b/tui/widgets/progress.py → bootstrap/tui/widgets/progress.py
@@ -20,7 +20,7 @@
 )
 from torch.utils.data.dataloader import DataLoader
 
-from tui import Task
+from bootstrap.tui import Task
 
 
 class DatasetProgressBar(Static):

diff --git a/src/base_tester.py b/src/base_tester.py
@@ -22,8 +22,8 @@
 from torch.utils.data import DataLoader
 from torchmetrics import MeanMetric
 
+from bootstrap.tui.training_ui import TrainingUI
 from src.base_trainer import BaseTrainer
-from tui.training_ui import TrainingUI
 from utils import to_cuda
 
 T = TypeVar("T")

diff --git a/src/base_trainer.py b/src/base_trainer.py
@@ -27,9 +27,9 @@
 from torch.utils.data import DataLoader
 from torchmetrics import MeanMetric
 
+from bootstrap.tui import Plot_BestModel
+from bootstrap.tui.training_ui import TrainingUI
 from conf import project as project_conf
-from tui import Plot_BestModel
-from tui.training_ui import TrainingUI
 from utils import to_cuda
 from utils.helpers import BestNModelSaver
 from utils.training import visualize_model_predictions