Reverted extension extraction to usage of pathlib

ami-iit · May 16, 2024 · 17d5028 · 17d5028
1 parent fdc1fec
commit 17d5028
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/rod/builder/primitives.py b/src/rod/builder/primitives.py
@@ -77,9 +77,9 @@ def __post_init__(self) -> None:
         """
 
         if isinstance(self.mesh_path, str):
-            extension = self.mesh_path.split(".")[-1]
+            extension = pathlib.Path(self.mesh_path).suffix
         elif isinstance(self.mesh_path, pathlib.Path):
-            extension = str(self.mesh_path).split(".")[-1]
+            extension = self.mesh_path.suffix
         else:
             raise TypeError(
                 f"Expected str or pathlib.Path for mesh_path, got {type(self.mesh_path)}"