fix unexpected key word (#34046)

4b888f67 · Jiangxinz · GitHub · 9e32a387 · 4b888f67
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

python/paddle/fluid/tests/unittests/test_dist_base.py python/paddle/fluid/tests/unittests/test_dist_base.py +2 -1

未找到文件。
--- a/python/paddle/fluid/tests/unittests/test_dist_base.py
+++ b/python/paddle/fluid/tests/unittests/test_dist_base.py
@@ -62,7 +62,8 @@ class TestDistRunnerBase(object):
                  batch_size=DEFAULT_BATCH_SIZE,
                  lr=0.1,
                  single_device=False,
-                  use_dgc=False):
+                  use_dgc=False,
+                  dist_strategy=None):
        raise NotImplementedError(
            "get_model should be implemented by child classes.")