From 38fce0d8265a117a39715955e66f3e2fddd5a7bb Mon Sep 17 00:00:00 2001 From: ShriyaPalsamudram Date: Wed, 13 Nov 2024 18:15:26 +0000 Subject: [PATCH] Apply isort and black reformatting Signed-off-by: ShriyaPalsamudram --- tests/collections/llm/test_fault_nvrx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: