From 4b262e441bacc2ba61e7b73623a7ce121b642f37 Mon Sep 17 00:00:00 2001 From: Joeri van Engelen Date: Wed, 14 Feb 2024 13:53:25 +0100 Subject: [PATCH] Type annotate merge_partitions --- xugrid/ugrid/ugrid1d.py | 2 +- xugrid/ugrid/ugrid2d.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/xugrid/ugrid/ugrid1d.py b/xugrid/ugrid/ugrid1d.py index a5f705d16..748450f7a 100644 --- 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 index 1055b72dc..4f8470b56 100644 --- 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.