fix: adding empty samples to project class

Loop3D · Feb 6, 2024 · bf100b2 · bf100b2
1 parent f0e82e2
commit bf100b2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/map2loop/project.py b/map2loop/project.py
@@ -139,7 +139,12 @@ def __init__(
         self.stratigraphic_column = StratigraphicColumn()
         self.deformation_history = DeformationHistory()
 
-        self.fault_orientations = None  # TODO set default values for all samplers
+        self.fault_orientations = pandas.DataFrame(
+            columns=["ID", "DIPDIR", "DIP", "X", "Y", "Z"]
+        )
+        self.fault_samples = pandas.DataFrame(columns=["ID", "X", "Y", "Z"])
+        self.fold_samples = pandas.DataFrame(columns=["ID", "X", "Y", "Z"])
+        self.geology_samples = pandas.DataFrame(columns=["ID", "X", "Y", "Z"])
         # Check for alternate config filenames in kwargs
         if "metadata_filename" in kwargs and config_filename == "":
             config_filename = kwargs["metadata_filename"]