made hpc signal handling optional

williamFalcon · Sep 5, 2019 · 2971859 · 2971859
1 parent a5dcf31
commit 2971859
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test_tube/hpc.py b/test_tube/hpc.py
@@ -134,12 +134,14 @@ def optimize_parallel_cluster_cpu(
             train_function,
             nb_trials,
             job_name,
+            enable_auto_resubmit=False,
             job_display_name=None
     ):
         if job_display_name is None:
             job_display_name = job_name
 
-        self.__optimize_parallel_cluster_internal(train_function, nb_trials, job_name, job_display_name, on_gpu=False)
+        self.__optimize_parallel_cluster_internal(train_function, nb_trials, job_name, job_display_name,
+                                                  enable_auto_resubmit, on_gpu=False)
 
     def __optimize_parallel_cluster_internal(
             self,