No clip to 0-1

haruishi43 · Jan 16, 2024 · 26cecbb · 26cecbb
1 parent 8d492b7
commit 26cecbb
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/equilib/cube2equi/numpy.py b/equilib/cube2equi/numpy.py
@@ -323,7 +323,7 @@ def run(
     out = (
         out.astype(horizon_dtype)
         if horizon_dtype == np.dtype(np.uint8)
-        else np.clip(out, 0.0, 1.0)
+        else np.clip(out, np.nanmin(out), np.nanmax(out))
     )
 
     return out
diff --git a/equilib/equi2cube/numpy.py b/equilib/equi2cube/numpy.py
@@ -206,7 +206,7 @@ def run(
     out = (
         out.astype(equi_dtype)
         if equi_dtype == np.dtype(np.uint8)
-        else np.clip(out, 0.0, 1.0)
+        else np.clip(out, np.nanmin(out), np.nanmax(out))
     )
 
     # reformat the output

diff --git a/equilib/equi2equi/numpy.py b/equilib/equi2equi/numpy.py
@@ -172,7 +172,7 @@ def run(
     out = (
         out.astype(src_dtype)
         if src_dtype == np.dtype(np.uint8)
-        else np.clip(out, 0.0, 1.0)
+        else np.clip(out, np.nanmin(out), np.nanmax(out))
     )
 
     return out
diff --git a/equilib/equi2pers/numpy.py b/equilib/equi2pers/numpy.py
@@ -226,7 +226,7 @@ def run(
     out = (
         out.astype(equi_dtype)
         if equi_dtype == np.dtype(np.uint8) or not clip_output
-        else np.clip(out, 0.0, 1.0)
+        else np.clip(out, np.nanmin(out), np.nanmax(out))
     )
 
     return out