Better fix for windows error

Lightning-Universe · Apr 22, 2021 · b1310e5 · b1310e5
1 parent 3bbfb16
commit b1310e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flash/data/data_module.py b/flash/data/data_module.py
@@ -80,7 +80,7 @@ def __init__(
 
         # TODO: figure out best solution for setting num_workers
         if num_workers is None:
-            if platform.system() == "Darwin" or os.getenv("FLASH_TESTING", "0") == "1":
+            if platform.system() == "Darwin" or platform.system() == "Windows":
                 num_workers = 0
             else:
                 num_workers = os.cpu_count()