Fix case with output_array=None

astropy · Sep 11, 2023 · dd766a0 · dd766a0
1 parent 6739839
commit dd766a0
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/reproject/mosaicking/coadd.py b/reproject/mosaicking/coadd.py
@@ -457,11 +457,13 @@ def reproject_and_coadd(
                     result.to_zarr(zarr_path)
                 result = da.from_zarr(zarr_path)
 
-            da.store(
-                result,
-                output_array,
-                compute=True,
-                scheduler="synchronous",
-            )
-
-            return output_array, None
+            if output_array is None:
+                return result.compute(scheduler="synchronous"), None
+            else:
+                da.store(
+                    result,
+                    output_array,
+                    compute=True,
+                    scheduler="synchronous",
+                )
+                return output_array, None