make sure accumulator is float64

spacetelescope · Nov 10, 2023 · d92d41e · d92d41e
1 parent 1c8d0a1
commit d92d41e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/drizzle/tests/test_drizzle.py b/drizzle/tests/test_drizzle.py
@@ -38,7 +38,10 @@ def centroid(image, size, center):
     xhi = min(xlo + size, image.shape[1])
 
     yx1 = np.mgrid[ylo:yhi, xlo:xhi, 1:2]
-    center = (yx1[..., 0] * image[ylo:yhi, xlo:xhi]).sum(axis=(1, 2))
+    center = (yx1[..., 0] * image[ylo:yhi, xlo:xhi]).sum(
+        axis=(1, 2),
+        dtype=np.float64,
+    )
 
     if center[2] == 0.0:
         return None