Add getter for vertex alignments (#511)

lukasturcani · Jul 19, 2023 · a514b46 · a514b46
1 parent 3d8e12a
commit a514b46
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/stk/_internal/topology_graphs/cage/cage.py b/src/stk/_internal/topology_graphs/cage/cage.py
@@ -1235,3 +1235,12 @@ def with_building_blocks(
         building_block_map: dict[BuildingBlock, BuildingBlock],
     ) -> Cage:
         return self.clone()._with_building_blocks(building_block_map)
+
+    def get_vertex_alignments(self) -> dict[int, int]:
+        """
+        Get the vertex alignments.
+
+        Returns:
+            The vertex alignments.
+        """
+        return dict(self._vertex_alignments)