fix spawn tests on windows

apache · Sep 17, 2019 · fe975d4 · fe975d4
1 parent 76ca2cc
commit fe975d4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/python/unittest/test_gluon_data.py b/tests/python/unittest/test_gluon_data.py
@@ -249,7 +249,10 @@ def test_multi_worker_forked_data_loader():
 @with_seed()
 def test_multi_worker_dataloader_release_pool():
     # will trigger too many open file if pool is not released properly
-    for _ in range(100):
+    if os.name == 'nt':
+        print('Skip for windows since spawn on windows is too expensive.')
+        return
+    for _ in range(10):
         A = np.random.rand(999, 2000)
         D = mx.gluon.data.DataLoader(A, batch_size=8, num_workers=8)
         the_iter = iter(D)