fix: fault orientation init as empty df rather than nan

Author: lachlangrose

LoopStructural/modelling/input/process_data.py

@@ -78,7 +78,7 @@ def __init__(
         self.contacts = contacts
         self._contact_orientations = None
         self.contact_orientations = contact_orientations
-        self._fault_orientations = None
+        self._fault_orientations = pd.DataFrame(columns=["X", "Y", "Z", "gx", "gy", "gz", "coord", "feature_name"])
         self.fault_orientations = fault_orientations
         self._fault_locations = None
         self.fault_locations = fault_locations
@@ -313,10 +313,11 @@ def fault_properties(self, fault_properties):
                 pts = self.fault_orientations.loc[
                     self.fault_orientations["feature_name"] == fname, ["gx", "gy", "gz"]
                 ]
-                fault_properties.loc[
-                    fname,
-                    ["avgNormalEasting", "avgNormalNorthing", "avgNormalAltitude"],
-                ] = np.nanmean(pts, axis=0)
+                if len(pts)>0:
+                    fault_properties.loc[
+                        fname,
+                        ["avgNormalEasting", "avgNormalNorthing", "avgNormalAltitude"],
+                    ] = np.nanmean(pts, axis=0)
         if (
             "avgSlipDirEasting" not in fault_properties.columns
             or "avgSlipDirNorthing" not in fault_properties.columns
@@ -399,7 +400,7 @@ def data(self):
             dataframes.append(self.contacts)
         if self.contact_orientations is not None:
             dataframes.append(self.contact_orientations)
-        if self.fault_orientations is not None:
+        if len(self.fault_orientations) > 0:
             dataframes.append(self.fault_orientations)
         if self.fault_locations is not None:
             dataframes.append(self.fault_locations)