bugfix

radio-astro-tools · Oct 13, 2024 · 25127dc · 25127dc
1 parent 63dfef9
commit 25127dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spectral_cube/spectral_cube.py b/spectral_cube/spectral_cube.py
@@ -1060,7 +1060,7 @@ def apply_function(self, function, axis=None, weights=None, unit=None,
         # check dtype first (for argmax/argmin)
         result = function(np.arange(3, dtype=self._data.dtype), **kwargs)
         if 'int' in str(result.dtype):
-            out = np.zeros([nz, nx, ny], dtype=dtype)
+            out = np.zeros([nz, nx, ny], dtype=result.dtype)
         else:
             out = np.empty([nz, nx, ny]) * np.nan