...

determined-ai · Jul 1, 2024 · 588934b · 588934b
1 parent 0bbaa22
commit 588934b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/e2e_tests/tests/cluster/test_master_restart_slurm.py b/e2e_tests/tests/cluster/test_master_restart_slurm.py
@@ -36,7 +36,7 @@ def test_master_restart_reattach_recover_experiment_slurm(
     managed_slurm_cluster_restarts: managed_slurm_cluster.ManagedSlurmCluster, downtime: int
 ) -> None:
     test_master_restart._test_master_restart_reattach_recover_experiment(
-        managed_slurm_cluster_restarts, downtime, max_workload_ticks=1000
+        managed_slurm_cluster_restarts, downtime, max_workload_ticks=500
     )
 
 

diff --git a/e2e_tests/tests/experiment/experiment.py b/e2e_tests/tests/experiment/experiment.py
@@ -412,7 +412,7 @@ def wait_for_at_least_n_trials(
 def wait_for_experiment_workload_progress(
     sess: api.Session, experiment_id: int, max_ticks: int = conf.MAX_TRIAL_BUILD_SECS
 ) -> None:
-    for _ in range(conf.MAX_TRIAL_BUILD_SECS):
+    for _ in range(max_ticks):
         trials = experiment_trials(sess, experiment_id)
         if len(trials) > 0:
             only_trial = trials[0]