Fix unit test

open-mmlab · Mar 8, 2023 · 15c5c2a · 15c5c2a
1 parent 4d21685
commit 15c5c2a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_utils/test_dl_utils/test_setup_env.py b/tests/test_utils/test_dl_utils/test_setup_env.py
@@ -13,8 +13,8 @@ def test_setup_multi_processes():
     sys_start_mehod = mp.get_start_method(allow_none=True)
     sys_cv_threads = cv2.getNumThreads()
     # pop and temp save system env vars
-    sys_omp_threads = os.environ.pop('OMP_NUM_THREADS', default=None)
-    sys_mkl_threads = os.environ.pop('MKL_NUM_THREADS', default=None)
+    sys_omp_threads = os.environ.pop('OMP_NUM_THREADS', None)
+    sys_mkl_threads = os.environ.pop('MKL_NUM_THREADS', None)
 
     # test distributed
     set_multi_processing(distributed=True)