Skip to content

Commit db80629

Browse files
committed
updated tests to use best model path as a property
1 parent 19a8794 commit db80629

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tests/integration_tests/bal/test_api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def test_kernel_selection(example_atoms, get_tmp_path, get_sample_input):
2323
_, params = initialize_model(model_config, inputs)
2424

2525
ckpt = {"model": {"params": params}, "epoch": 0}
26-
best_dir = model_config.data.best_model_path()
26+
best_dir = model_config.data.best_model_path
2727
checkpoints.save_checkpoint(
2828
ckpt_dir=best_dir,
2929
target=ckpt,

tests/integration_tests/md/test_md.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ def test_ase_calc(get_tmp_path):
147147
)
148148
ckpt = {"model": {"params": params}, "epoch": 0}
149149

150-
best_dir = model_config.data.best_model_path()
150+
best_dir = model_config.data.best_model_path
151151
checkpoints.save_checkpoint(
152152
ckpt_dir=best_dir,
153153
target=ckpt,

0 commit comments

Comments
 (0)