try something to patch test ssda

aramis-lab · Oct 18, 2024 · 40322fc · 40322fc
1 parent 12b1463
commit 40322fc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/tests/test_train_ae.py b/tests/test_train_ae.py
@@ -106,6 +106,10 @@ def test_train_ae(cmdopt, tmp_path, test_name):
             base_dir=base_dir,
             no_gpu=cmdopt["no-gpu"],
             adapt_base_dir=cmdopt["adapt-base-dir"],
+        )
+        json_data_out = modify_maps(
+            maps=json_data_out,
+            base_dir=base_dir,
             ssda=True,
         )
     assert json_data_out == json_data_ref  # ["mode"] == mode

diff --git a/tests/test_train_cnn.py b/tests/test_train_cnn.py
@@ -127,6 +127,11 @@ def test_train_cnn(cmdopt, tmp_path, test_name):
             adapt_base_dir=cmdopt["adapt-base-dir"],
             ssda=True,
         )
+        json_data_out = modify_maps(
+            maps=json_data_out,
+            base_dir=base_dir,
+            ssda=True,
+        )
     assert json_data_out == json_data_ref  # ["mode"] == mode
 
     assert compare_folders(