From 7d0787ad0a6592b07cb3596b00828c739d627fa6 Mon Sep 17 00:00:00 2001 From: domfournier Date: Wed, 5 Feb 2025 18:38:15 -0800 Subject: [PATCH] Avoid empty blocks --- simpeg/dask/utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/simpeg/dask/utils.py b/simpeg/dask/utils.py index 10188b7bc0..c8c74168a0 100644 --- a/simpeg/dask/utils.py +++ b/simpeg/dask/utils.py @@ -77,5 +77,8 @@ def get_parallel_blocks( flatten_blocks += block chunks = np.array_split(np.arange(len(flatten_blocks)), cpu_count()) - return [[flatten_blocks[i] for i in chunk] for chunk in chunks] + return [ + [flatten_blocks[i] for i in chunk] for chunk in chunks if len(chunk) > 0 + ] + return blocks