Skip to content
Snippets Groups Projects
Commit 1fb90fd3 authored by leufen1's avatar leufen1
Browse files

test update

parent c3bca050
No related branches found
No related tags found
3 merge requests!192include Develop,!191Resolve "release v1.1.0",!182Resolve "Mixed sampling types"
Pipeline #51132 passed
......@@ -61,6 +61,7 @@ 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")
if not os.path.exists(path):
os.makedirs(path)
obj.data_store.set("experiment_path", path, "general")
os.makedirs(batch_path)
......@@ -170,6 +171,7 @@ class TestTraining:
@pytest.fixture
def ready_to_init(self, data_collection, model, callbacks, path, model_path, batch_path):
if not os.path.exists(path):
os.makedirs(path)
os.makedirs(model_path)
obj = RunEnvironment()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment