From aba8676d53bc76e4bcb1209445508e50083a5342 Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Mon, 4 Jul 2022 21:33:37 +0200 Subject: [PATCH] Make cross platform --- great_ai/utilities/parallel_map/get_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/great_ai/utilities/parallel_map/get_config.py b/great_ai/utilities/parallel_map/get_config.py index b053848..4f96cdc 100644 --- a/great_ai/utilities/parallel_map/get_config.py +++ b/great_ai/utilities/parallel_map/get_config.py @@ -19,7 +19,7 @@ def get_config( is_input_sequence = hasattr(input_values, "__len__") if concurrency is None: - concurrency = len(os.sched_getaffinity(0)) + concurrency = os.cpu_count() or 1 assert concurrency >= 1, "At least one mapper process has to be created" if chunk_size is None: