fix: 🐛 faults where feature name given not fault_name were crashing

Loop3D · May 4, 2022 · dae9e92 · dae9e92
1 parent 92da55f
commit dae9e92
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/LoopStructural/modelling/input/process_data.py b/LoopStructural/modelling/input/process_data.py
@@ -619,7 +619,10 @@ def fault_orientations(self, fault_orientations):
             fault_orientations[["gx", "gy", "gz"]] = strike_dip_vector(
                 fault_orientations["strike"], fault_orientations["dip"]
             )
-        fault_orientations["feature_name"] = fault_orientations["fault_name"]
+        if "feature_name" not in fault_orientations.columns and "fault_name" in fault_orientations.columns:
+            fault_orientations["feature_name"] = fault_orientations["fault_name"]
+        if "feature_name" not in fault_orientations.columns:
+            raise ValueError("Fault orientation data must contain feature_name or fault_name")
         self._fault_orientations = fault_orientations[
             ["X", "Y", "Z", "gx", "gy", "gz", "coord", "feature_name"]
         ]