BUG: fix well export to rms for masked numpy with Nan

equinor · Jan 7, 2025 · 5daa411 · 5daa411
1 parent 1af8dbb
commit 5daa411
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/xtgeo/well/_well_roxapi.py b/src/xtgeo/well/_well_roxapi.py
@@ -10,7 +10,7 @@
 
 from xtgeo.common import XTGeoDialog, null_logger
 from xtgeo.common._xyz_enum import _AttrName, _AttrType
-from xtgeo.common.constants import UNDEF_INT_LIMIT, UNDEF_LIMIT
+from xtgeo.common.constants import UNDEF, UNDEF_INT, UNDEF_INT_LIMIT, UNDEF_LIMIT
 from xtgeo.roxutils import RoxUtils
 
 if TYPE_CHECKING:
@@ -189,9 +189,11 @@ def _store_log_in_roxapi(self, lrun: Any, logname: str) -> None:
 
     isdiscrete = False
     xtglimit = UNDEF_LIMIT
+    apply_undef = UNDEF
     if self.wlogtypes[logname] == _AttrType.DISC.value:
         isdiscrete = True
         xtglimit = UNDEF_INT_LIMIT
+        apply_undef = UNDEF_INT
 
     store_logname = logname
 
@@ -215,7 +217,10 @@ def _store_log_in_roxapi(self, lrun: Any, logname: str) -> None:
 
     vals = np.ma.masked_invalid(self.get_dataframe(copy=False)[logname].values)
     vals = np.ma.masked_greater(vals, xtglimit)
-    vals = vals.astype(usedtype)
+
+    # to avoid that Nans behind the masks trigger RuntimeWarning, they are converted to
+    # the apply_undef value priot to astype() (see issue 1283)
+    vals = np.nan_to_num(vals, nan=apply_undef).astype(usedtype)
     thelog.set_values(vals)
 
     if isdiscrete: