BUG: Invoke the right image type

pmgbergen · Jun 25, 2024 · 5dc6c87 · 5dc6c87
1 parent 0ca0d61
commit 5dc6c87
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/darsia/utils/standard_images.py b/src/darsia/utils/standard_images.py
@@ -29,9 +29,10 @@ def zeros_like(
     if dtype is None:
         dtype = image.dtype
     if mode == "shape":
-        return darsia.Image(np.zeros(image.shape, dtype=dtype), metadata=image.metadata)
+        ImageType = type(image)
+        return ImageType(np.zeros(image.shape, dtype=dtype), metadata=image.metadata)
     elif mode == "voxels":
-        return darsia.Image(
+        return darsia.ScalarImage(
             np.zeros(image.num_voxels, dtype=dtype), metadata=image.metadata
         )
 
@@ -56,8 +57,9 @@ def ones_like(
     if dtype is None:
         dtype = image.dtype
     if mode == "shape":
-        return darsia.Image(np.ones(image.shape, dtype=dtype), metadata=image.metadata)
+        ImageType = type(image)
+        return ImageType(np.ones(image.shape, dtype=dtype), metadata=image.metadata)
     elif mode == "voxels":
-        return darsia.Image(
+        return darsia.ScalarImage(
             np.ones(image.num_voxels, dtype=dtype), metadata=image.metadata
         )