TST: Fix test asserts that check for log files

vocalpy · Dec 2, 2022 · a8f3d18 · a8f3d18
1 parent 9f6017d
commit a8f3d18
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/test_core/test_learncurve.py b/tests/test_core/test_learncurve.py
@@ -38,7 +38,7 @@ def learncurve_output_matches_expected(cfg, model_config_map, results_path):
                 assert model_path.exists()
 
                 tensorboard_log = sorted(
-                    model_path.glob(f"events.out.tfevents.*{model_name}")
+                    replicate_path.glob(f"lightning_logs/**/*events*")
                 )
                 assert len(tensorboard_log) == 1
 

diff --git a/tests/test_core/test_train.py b/tests/test_core/test_train.py
@@ -19,7 +19,9 @@ def train_output_matches_expected(cfg, model_config_map, results_path):
         model_path = results_path.joinpath(model_name)
         assert model_path.exists()
 
-        tensorboard_log = sorted(model_path.glob(f"events.out.tfevents.*{model_name}"))
+        tensorboard_log = sorted(
+            model_path.glob(f"lightning_logs/**/*events*")
+        )
         assert len(tensorboard_log) == 1
 
         checkpoints_path = model_path.joinpath("checkpoints")