fix fleet dist infer ut

Change-Id: Id9ce0410b5976b18c72a957af017923f655ed494
PaddlePaddle · May 26, 2021 · 6c943a4 · 6c943a4 · paddle-bot-old · May 26, 2021
1 parent c7aabfb
commit 6c943a4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_dist_fleet_geo.py b/python/paddle/fluid/tests/unittests/test_dist_fleet_geo.py
@@ -42,7 +42,9 @@ def check_with_place(self,
             "PYTHONPATH": os.getenv("PYTHONPATH", ""),
             "LD_LIBRARY_PATH": os.getenv("LD_LIBRARY_PATH", ""),
             "FLAGS_rpc_deadline": "5000",  # 5sec to fail fast
-            "http_proxy": ""
+            "http_proxy": "",
+            "LOG_DIRNAME": "/tmp",
+            "LOG_PREFIX": self.__class__.__name__,
         }
 
         required_envs.update(need_envs)
@@ -55,7 +57,7 @@ def check_with_place(self,
 
     def test_dist_train(self):
         self.check_with_place(
-            "dist_fleet_ctr.py", delta=1e-5, check_error_log=True)
+            "dist_fleet_ctr.py", delta=1e-5, check_error_log=False)
 
 
 class TestGeoSgdTranspiler(unittest.TestCase):