From a2c92c94d9f5a0dcec83b5468e5c029ede980578 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Moritz=20R=2E=20Sch=C3=A4fer?= Date: Fri, 10 Nov 2023 22:10:02 +0100 Subject: [PATCH] use best model dir directly in tests --- tests/integration_tests/md/test_md.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/tests/integration_tests/md/test_md.py b/tests/integration_tests/md/test_md.py index abca6214..6a0249d1 100644 --- a/tests/integration_tests/md/test_md.py +++ b/tests/integration_tests/md/test_md.py @@ -80,11 +80,8 @@ def test_run_md(get_tmp_path): ) ckpt = {"model": {"params": params}, "epoch": 0} - best_dir = os.path.join( - model_config.data.directory, model_config.data.experiment, "best" - ) checkpoints.save_checkpoint( - ckpt_dir=best_dir, + ckpt_dir=model_config.data.best_model_path, target=ckpt, step=0, overwrite=True, @@ -147,9 +144,8 @@ def test_ase_calc(get_tmp_path): ) ckpt = {"model": {"params": params}, "epoch": 0} - best_dir = model_config.data.best_model_path checkpoints.save_checkpoint( - ckpt_dir=best_dir, + ckpt_dir=model_config.data.best_model_path, target=ckpt, step=0, overwrite=True,