fix adamax test (#51366)

f759cf0f · 姜永久 · GitHub · d9de3ef6 · f759cf0f
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

python/paddle/fluid/tests/unittests/test_adamax_op.py python/paddle/fluid/tests/unittests/test_adamax_op.py +2 -2

未找到文件。
--- a/python/paddle/fluid/tests/unittests/test_adamax_op.py
+++ b/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):