Skip to content

Commit 4ebb112

Browse files
hanwen-clusterhanwen-pcluste
authored andcommitted
[integ-tests] Reduce test_slurm_accounting_external_dbd sporadic failure
Signed-off-by: Hanwen <hanwenli@amazon.com>
1 parent b7347d5 commit 4ebb112

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/integration-tests/tests/schedulers/test_slurm_accounting.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,9 @@ def _test_jobs_get_recorded(scheduler_commands):
123123
job_id = scheduler_commands.assert_job_submitted(job_submission_output)
124124
logging.info(" Submitted Job ID: %s", job_id)
125125
scheduler_commands.wait_job_completed(job_id)
126-
_assert_job_completion_recorded_in_accounting(job_id, scheduler_commands)
126+
retry(stop_max_attempt_number=5, wait_fixed=seconds(5))(_assert_job_completion_recorded_in_accounting)(
127+
job_id, scheduler_commands
128+
)
127129

128130

129131
def _assert_job_completion_recorded_in_accounting(job_id, scheduler_commands, clusters=None):

0 commit comments

Comments
 (0)