Commit 1fb90fd3 authored by lukas leufen's avatar lukas leufen

test update

parent c3bca050
Pipeline #51132 passed with stages
in 8 minutes and 18 seconds
......@@ -61,7 +61,8 @@ class TestTraining:
obj.data_store.set("data_collection", mock.MagicMock(return_value="mock_train_gen"), "general.train")
obj.data_store.set("data_collection", mock.MagicMock(return_value="mock_val_gen"), "general.val")
obj.data_store.set("data_collection", mock.MagicMock(return_value="mock_test_gen"), "general.test")
os.makedirs(path)
if not os.path.exists(path):
os.makedirs(path)
obj.data_store.set("experiment_path", path, "general")
os.makedirs(batch_path)
obj.data_store.set("batch_path", batch_path, "general")
......@@ -170,7 +171,8 @@ class TestTraining:
@pytest.fixture
def ready_to_init(self, data_collection, model, callbacks, path, model_path, batch_path):
os.makedirs(path)
if not os.path.exists(path):
os.makedirs(path)
os.makedirs(model_path)
obj = RunEnvironment()
obj.data_store.set("data_collection", data_collection, "general.train")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment