From a547d5bda5f4b85bf5571b4f770fa9062f5a6d8b Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Fri, 25 Nov 2022 09:02:10 -0800 Subject: [PATCH] [AnyPrecisionAdamW] test fix (#20454) --- tests/trainer/test_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py index 19016640c9d6d4..dd3ac61ea11af9 100644 --- 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, )