From 521fe06e7b46e14e018a88a5a8a74ee052542697 Mon Sep 17 00:00:00 2001 From: Ceferino Patino Date: Thu, 10 Oct 2024 22:05:27 -0500 Subject: [PATCH] Update slurm_handler.py --- hyperparameter_tuner/slurm_handler.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/hyperparameter_tuner/slurm_handler.py b/hyperparameter_tuner/slurm_handler.py index c4b69e8..1320d29 100644 --- a/hyperparameter_tuner/slurm_handler.py +++ b/hyperparameter_tuner/slurm_handler.py @@ -70,6 +70,10 @@ def create_objective( logger.error('Error submitting SLURM job') raise + while not os.path.isfile(results_path): + print(f"Waiting for {results_path} to be created...") + time.sleep(5) + while True: df = pd.read_csv(results_path)