fixed model loading test

apax-hub · Dec 28, 2023 · 8e0cd03 · 8e0cd03
1 parent bbbbd76
commit 8e0cd03
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/integration_tests/md/test_model_loading.py b/tests/integration_tests/md/test_model_loading.py
@@ -21,7 +21,7 @@ def test_model_loading(get_tmp_path, get_sample_input):
 
     ckpt = {"model": {"params": params}, "epoch": 0}
     ckpt_dir1 = pathlib.Path("models/apax_dummy/best")
-    ckpt_dir2 = pathlib.Path("../models/apax_dummy/best").resolve()
+    ckpt_dir2 = pathlib.Path("../models/apax_dummy/best")
     ckpt_dir3 = pathlib.Path("models/best")
 
     ckpt_dirs = [ckpt_dir1, ckpt_dir2, ckpt_dir3]
@@ -32,7 +32,7 @@ def test_model_loading(get_tmp_path, get_sample_input):
         model_config.dump_config(ckpt_dir.parent)
 
         checkpoints.save_checkpoint(
-            ckpt_dir=ckpt_dir,
+            ckpt_dir=ckpt_dir.resolve(),
             target=ckpt,
             step=0,
             overwrite=True,