polish

7 years ago · 12ea358c06
parent 6e26bfe8ef
commit 12ea358c06
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/tests/unittests/dist_transformer.py
+++ b/python/paddle/fluid/tests/unittests/dist_transformer.py
@ -172,7 +172,7 @@ def get_transpiler(trainer_id, main_program, pserver_endpoints, trainers):
    return t


-class DistTransformer2x2:
+class DistTransformer2x2(object):
    def run_pserver(self, pserver_endpoints, trainers, current_endpoint,
                    trainer_id):
        get_model()