Merge pull request #6 from mpaillassa/type_fix

Float cast fix
mpaillassa · Jan 13, 2022 · e829018 · e829018
2 parents e03c947 + c406160
commit e829018
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/maximask_and_maxitrack/utils.py b/maximask_and_maxitrack/utils.py
@@ -282,6 +282,9 @@ def clean_grid(grid):
 def image_norm(im):
     """Image preprocessing and normalization"""
 
+    # safety cast
+    im = im.astype(np.float32)
+
     # preprocessing
     np.place(im, np.isnan(im), 0)
     np.place(im, im > 80000, 80000)

diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
 
 setup(
     name="MaxiMask",
-    version="1.3.0",
+    version="1.3.1",
     author="Maxime Paillassa",
     author_email="[email protected]",
     description="Convolutional neural networks to detect contaminants in astronomical images.",