From eebdb3726b3bf43b16a7fca1f6f042654b2b67dc Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Mon, 6 Nov 2023 22:59:19 -0500 Subject: [PATCH] throw error if training_iter0_model_path is not found (#1391) Fix #622. --------- Signed-off-by: Jinzhe Zeng Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> --- dpgen/generator/run.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/dpgen/generator/run.py b/dpgen/generator/run.py index 034918efc..8e6da1d7d 100644 --- a/dpgen/generator/run.py +++ b/dpgen/generator/run.py @@ -632,7 +632,10 @@ def make_train(iter_index, jdata, mdata): % numb_models ) for ii in range(len(iter0_models)): - old_model_files = glob.glob(os.path.join(iter0_models[ii], "model.ckpt*")) + old_model_path = os.path.join(iter0_models[ii], "model.ckpt*") + old_model_files = glob.glob(old_model_path) + if not len(old_model_files): + raise FileNotFoundError(f"{old_model_path} not found!") _link_old_models(work_path, old_model_files, ii) copied_models = next( (