diff --git a/flash/data/data_module.py b/flash/data/data_module.py index 8952ef5b17..fea3e7bbc3 100644 --- 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()