Type annotate merge_partitions

Deltares · Feb 14, 2024 · 4b262e4 · 4b262e4
1 parent d9b3494
commit 4b262e4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/xugrid/ugrid/ugrid1d.py b/xugrid/ugrid/ugrid1d.py
@@ -630,7 +630,7 @@ def contract_vertices(self, indices: IntArray) -> "Ugrid1d":
         )
 
     @staticmethod
-    def merge_partitions(grids: Sequence["Ugrid1d"]) -> "Ugrid1d":
+    def merge_partitions(grids: Sequence["Ugrid1d"]) -> tuple["Ugrid1d", dict[str, np.array]]:
         """
         Merge grid partitions into a single whole.
 

diff --git a/xugrid/ugrid/ugrid2d.py b/xugrid/ugrid/ugrid2d.py
@@ -1505,7 +1505,7 @@ def partition(self, n_part: int):
         return [self.topology_subset(index) for index in indices]
 
     @staticmethod
-    def merge_partitions(grids: Sequence["Ugrid2d"]) -> "Ugrid2d":
+    def merge_partitions(grids: Sequence["Ugrid2d"]) -> tuple["Ugrid2d", dict[str, np.array]]:
         """
         Merge grid partitions into a single whole.