[AnyPrecisionAdamW] test fix (#20454)

a547d5bd · Stas Bekman · GitHub · a1d4563f · a547d5bd
Unverified Commit a547d5bd authored Nov 25, 2022 by Stas Bekman Committed by GitHub Nov 25, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/trainer/test_trainer.py tests/trainer/test_trainer.py +1 -1

No files found.
--- a/tests/trainer/test_trainer.py
+++ b/tests/trainer/test_trainer.py
@@ -2332,7 +2332,7 @@ if is_torch_available():

        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,
            )