fix adamax test (#51366)

python/paddle/fluid/tests/unittests/test_adamax_op.py

@@ -27,6 +27,7 @@ def adamx_wrapper(
     moment,
     inf_norm,
     beta1_pow=None,
+    master_weight=None,
     beta1=0.78,
     beta2=0.899,
     epsilon=1e-5,
@@ -38,13 +39,12 @@ def adamx_wrapper(
         moment,
         inf_norm,
         beta1_pow,
+        master_weight,
         beta1,
         beta2,
         epsilon,
     )
 
-import paddle
-
 
 class TestAdamaxOp1(OpTest):
     def setUp(self):