🎨 Auto-generated directory tree for repository in Architecture.md

DSGT-DLP · May 12, 2024 · fd62c6c · fd62c6c
1 parent 692e8c8
commit fd62c6c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/.github/Architecture.md b/.github/Architecture.md
@@ -34,6 +34,7 @@
 |  |  |  |  |- 📜 trainer.py
 |  |  |  |  |- 📜 criterion.py
 |  |  |  |  |- 📜 dl_model.py : torch model based on user specifications from drag and drop
+|  |  |  |  |- 📜 train_types.py
 |  |  |  |  |- 📜 dataset.py : read in the dataset through URL or file upload
 |  |  |  |  |- 📜 __init__.py
 |  |  |  |  |- 📜 Dockerfile

diff --git a/training/training/core/celery/train_types.py b/training/training/core/celery/train_types.py
@@ -26,7 +26,6 @@ class TrainResultsData(Schema):
     dataSource: DATA_SOURCE
     status: TRAIN_STATUS
     created: datetime.date
-    step: str
     uid: str