cast to np.uint8 the maxorder

cds-astro · Sep 5, 2024 · d6cf645 · d6cf645
1 parent 2cc5d4a
commit d6cf645
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 26 deletions.
diff --git a/notebooks/from_image_with_mask.ipynb b/notebooks/from_image_with_mask.ipynb
diff --git a/python/mocpy/moc/moc.py b/python/mocpy/moc/moc.py
@@ -522,6 +522,7 @@ def from_fits_image(cls, hdu, max_norder, mask=None):
         """
         # Only take the first HDU
         header = hdu.header
+        max_norder = np.uint8(max_norder)
 
         # Compute a WCS from the header of the image
         w = wcs.WCS(header)
@@ -573,10 +574,13 @@ def from_fits_image(cls, hdu, max_norder, mask=None):
                 u.rad,
             )  # in steradians
 
+            # Division by 0 case
             if px_sky_area == 0.0:
                 healpix_order_computed = False
             else:
-                depth_px = np.floor(np.log2(np.pi / (3.0 * px_sky_area)) / 2.0)
+                depth_px = np.uint8(
+                    np.floor(np.log2(np.pi / (3.0 * px_sky_area)) / 2.0),
+                )
                 max_norder = min(max_norder, depth_px)
         else:
             healpix_order_computed = False