Allow multiple output functions to just return a tuple

cubed-dev · Sep 16, 2024 · 18ad533 · 18ad533
1 parent 887d802
commit 18ad533
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/cubed/primitive/blockwise.py b/cubed/primitive/blockwise.py
@@ -89,8 +89,10 @@ def apply_blockwise(out_coords: List[int], *, config: BlockwiseSpec) -> None:
         args.append(arg)
 
     results = config.function(*args)
-    # if blockwise function is a regular function (not a generator) then make it iterable
-    if not inspect.isgeneratorfunction(config.function):
+    # if blockwise function is a regular function (not a generator) that doesn't return multiple values then make it iterable
+    if not inspect.isgeneratorfunction(config.function) and not isinstance(
+        results, tuple
+    ):
         results = (results,)
     for i, result in enumerate(results):
         out_chunk_key = key_to_slices(