Extrapolate air values for referenced petrophysics model

Author: domfournier

simpeg_drivers/components/models.py

@@ -17,7 +17,7 @@
 from geoapps_utils.utils.importing import GeoAppsError
 from geoapps_utils.utils.numerical import weighted_average
 from geoh5py.data import Data, FloatData, NumericData
-from geoh5py.data.data_type import GeometricDataValueMapType
+from geoh5py.data.data_type import GeometricDataValueMapType, ReferencedValueMapType
 from geoh5py.objects import ObjectBase
 from simpeg.utils.mat_utils import (
     dip_azimuth2cartesian,
@@ -662,7 +662,9 @@ def obj_2_mesh(data: Data, destination: ObjectBase) -> np.ndarray:
 
         values = data.values.astype(float)
 
-        if isinstance(data.entity_type, GeometricDataValueMapType):
+        if isinstance(
+            data.entity_type, GeometricDataValueMapType | ReferencedValueMapType
+        ):
             values[values == 0] = np.nan
 
         full_vector = weighted_average(xyz_in, xyz_out, [values], n=1)[0]