Merge pull request #19 from class255/master

No clip to 0-1
haruishi43 · Jan 16, 2024 · dc352ee · dc352ee
2 parents 8d492b7 + 26cecbb
commit dc352ee
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