typo pytest description string

huggingface · Apr 19, 2024 · 2de09d3 · 2de09d3
1 parent 6c058de
commit 2de09d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_optimizer_params_groups.py b/tests/test_optimizer_params_groups.py
@@ -489,7 +489,7 @@ def optimizer_builder(inp_param_groups):
             torch.testing.assert_close(expected_fc2_weight, updated_fc2_weight)
 
 
-@pytest.mark.skipif(available_gpus() < 2, reason="Testing ddp_hook_allreduce requires at least 2 gpus")
+@pytest.mark.skipif(available_gpus() < 2, reason="Testing requires at least 2 gpus")
 @pytest.mark.parametrize("half_precision", [torch.float16, torch.bfloat16])
 @pytest.mark.parametrize("accumulation_steps", [1, 10])
 @rerun_if_address_is_in_use()