diff --git a/tests/collections/llm/test_fault_nvrx.py b/tests/collections/llm/test_fault_nvrx.py index b5eb1201bbf3b..c90a099e5bf2a 100644 --- a/tests/collections/llm/test_fault_nvrx.py +++ b/tests/collections/llm/test_fault_nvrx.py @@ -98,7 +98,7 @@ def main(): run_plugins: list[run.Plugin] = [FaultTolerancePlugin(num_in_process_restarts=1, num_job_retries_on_failure=0)] pretrain_recipe.trainer.callbacks = [ run.Config(TimingCallback), - straggler_det_callback(straggler_report_time_interval=0.5) + straggler_det_callback(straggler_report_time_interval=0.5), ] if args.crash_step: