fix: returning two x coordinates

borgesaugusto · Jul 6, 2023 · d2f36a9 · d2f36a9
1 parent e33fbfd
commit d2f36a9
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/seapipy/lattice_class.py b/seapipy/lattice_class.py
@@ -200,10 +200,9 @@ def get_coordinates(vertices: dict) -> tuple:
         :return: X and Y coordinates
         :rtype: tuple
         """
-        x_coordinates = [x[0] for x in vertices.values()]
-        y_coordinates = [y[0] for y in vertices.values()]
+        x_coordinates, y_coordinates = zip(*vertices.values())
 
-        return x_coordinates, y_coordinates
+        return list(x_coordinates), list(y_coordinates)
 
     @staticmethod
     def get_edge_centroid(edge_id: int, vertices: list, edges: list) -> tuple: