Adding PostfitError to context manager

NNPDF · Apr 15, 2021 · 523068e · 523068e
1 parent 1c493ed
commit 523068e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/validphys2/src/validphys/scripts/postfit.py b/validphys2/src/validphys/scripts/postfit.py
@@ -130,7 +130,7 @@ def _postfit(results: str, nrep: int, chi2_threshold: float, arclength_threshold
     with tempfile_cleaner(
         root=result_path,
         exit_func=shutil.move,
-        exc=KeyboardInterrupt,
+        exc=(KeyboardInterrupt, PostfitError),
         prefix="postfit_work_deleteme_",
         dst=final_postfit_path,
     ) as postfit_path: