add hydra config.

remove click. add launch script. add test dir. switch from fashion mnist to generic.
2023-01-26 07:25:07 -08:00
parent 404e39206b
commit 1f13224c4f
11 changed files with 123 additions and 43 deletions
--- a/test/init.py
+++ b/test/init.py
--- a/test/test_pipeline.py
+++ b/test/test_pipeline.py
@@ -0,0 +1,10 @@
+from src.model.linear import DNN
+from src.data import GenericDataset
+import os
+
+
+def test_size_of_dataset():
+    features = 40
+    os.environ["INPUT_FEATURES"] = str(features)
+    dataset = GenericDataset()
+    assert len(dataset[0][0]) == features