update: add default termination for slab builder

Exabyte-io · Sep 26, 2024 · 8c18184 · 8c18184
1 parent 7e3f4f5
commit 8c18184
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/py/mat3ra/made/tools/build/slab/__init__.py b/src/py/mat3ra/made/tools/build/slab/__init__.py
@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Optional
 
 from mat3ra.made.material import Material
 from .builders import SlabBuilder, SlabSelectorParameters
@@ -10,6 +10,7 @@ def get_terminations(configuration: SlabConfiguration) -> List[Termination]:
     return SlabBuilder().get_terminations(configuration)
 
 
-def create_slab(configuration: SlabConfiguration, termination: Termination) -> Material:
+def create_slab(configuration: SlabConfiguration, termination: Optional[Termination] = None) -> Material:
     builder = SlabBuilder()
+    termination = termination or builder.get_terminations(configuration)[0]
     return builder.get_material(configuration, selector_parameters=SlabSelectorParameters(termination=termination))