fix: Add timeout for locking post_run_csv

automl · Nov 29, 2024 · c7592c1 · c7592c1
1 parent c0b5b77
commit c7592c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neps/status/status.py b/neps/status/status.py
@@ -299,7 +299,7 @@ def _save_data_to_csv(
         config_data_df: The DataFrame containing configuration data.
         run_data_df: The DataFrame containing additional run data.
     """
-    with locker(poll=2):
+    with locker(poll=2, timeout=120):
         try:
             pending_configs = run_data_df.loc["num_pending_configs", "value"]
             pending_configs_with_worker = run_data_df.loc[