[AnyPrecisionAdamW] test fix (#20454)

huggingface · Nov 25, 2022 · a547d5b · a547d5b
1 parent a1d4563
commit a547d5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
@@ -2332,7 +2332,7 @@ def hp_name(trial):
 
         optim_test_params.append(
             (
-                TrainingArguments(OptimizerNames.ADAMW_APEX_FUSED, output_dir="None"),
+                TrainingArguments(optim=OptimizerNames.ADAMW_APEX_FUSED, output_dir="None"),
                 apex.optimizers.FusedAdam,
                 default_adam_kwargs,
             )