ci fix log dir

Oneflow-Inc · Oct 21, 2020 · 5bb3701 · 5bb3701
1 parent 112b786
commit 5bb3701
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/ci/test/parallel_run.py b/ci/test/parallel_run.py
@@ -51,7 +51,7 @@ def available_slots():
                         os.environ,
                         CUDA_VISIBLE_DEVICES=cuda_visible_devices,
                         ONEFLOW_TEST_CTRL_PORT=str(find_free_port()),
-                        ONEFLOW_TEST_LOG_DIR=("./unittest-log-" + str(uuid.uuid4())),
+                        ONEFLOW_TEST_LOG_DIR=("./unittest-log/" + str(uuid.uuid4())),
                     ),
                     shell=True,
                 )

diff --git a/oneflow/python/framework/unittest.py b/oneflow/python/framework/unittest.py
@@ -151,16 +151,17 @@ def setUp(self):
                 atexit.register(oneflow.deprecated.delete_worker)
                 _unittest_worker_initilized = True
 
+        log_dir = os.getenv("ONEFLOW_TEST_LOG_DIR")
+        if log_dir:
+            oneflow.env.log_dir(log_dir)
+
         if _unittest_env_initilized == False:
             oneflow.env.init()
             _unittest_env_initilized = True
 
         oneflow.clear_default_session()
         oneflow.enable_eager_execution(eager_execution_enabled())
         oneflow.experimental.enable_typing_check(typing_check_enabled())
-        log_dir = os.getenv("ONEFLOW_TEST_LOG_DIR")
-        if log_dir:
-            oneflow.env.log_dir(log_dir)
 
 
 def skip_unless(n, d):