From f0a11f17dc1e62ae0a61c783d5b042194b63a645 Mon Sep 17 00:00:00 2001 From: Tom McTiernan Date: Wed, 22 Jun 2022 15:25:43 +0100 Subject: [PATCH] Skip unnecessary file checks --- submitit/core/submission.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/submitit/core/submission.py b/submitit/core/submission.py index 4d9f65f1..7fed7535 100644 --- a/submitit/core/submission.py +++ b/submitit/core/submission.py @@ -41,8 +41,9 @@ def process_job(folder: Union[Path, str]) -> None: logger.info(f"Loading pickle: {paths.submitted_pickle}") wait_time = 60 for _ in range(wait_time): - if not paths.submitted_pickle.exists(): - time.sleep(1) + if paths.submitted_pickle.exists(): + break + time.sleep(1) if not paths.submitted_pickle.exists(): raise RuntimeError( f"Waited for {wait_time} seconds but could not find submitted jobs in path:\n{paths.submitted_pickle}"