diff --git a/xugrid/ugrid/ugrid1d.py b/xugrid/ugrid/ugrid1d.py index 56ca7e22..6bb0da8b 100644 --- a/xugrid/ugrid/ugrid1d.py +++ b/xugrid/ugrid/ugrid1d.py @@ -731,14 +731,10 @@ def create_data_array(self, data: ArrayLike, facet: str) -> "xugrid.UgridDataArr ------- uda: UgridDataArray """ - match facet: - case "node": - dimension = self.node_dimension - case "edge": - dimension = self.edge_dimension - case _: - raise ValueError( - f"Invalid facet: {facet}. Must be one of: node, edge face." - ) - + if facet == "node": + dimension = self.node_dimension + elif facet == "edge": + dimension = self.edge_dimension + else: + raise ValueError(f"Invalid facet: {facet}. Must be one of: node, edge.") return self._create_data_array(data, dimension) diff --git a/xugrid/ugrid/ugrid2d.py b/xugrid/ugrid/ugrid2d.py index 3c026233..9013cdec 100644 --- a/xugrid/ugrid/ugrid2d.py +++ b/xugrid/ugrid/ugrid2d.py @@ -2264,16 +2264,12 @@ def create_data_array(self, data: ArrayLike, facet: str) -> "xugrid.UgridDataArr ------- uda: UgridDataArray """ - match facet: - case "node": - dimension = self.node_dimension - case "edge": - dimension = self.edge_dimension - case "face": - dimension = self.face_dimension - case _: - raise ValueError( - f"Invalid facet: {facet}. Must be one of: node, edge face." - ) - + if facet == "node": + dimension = self.node_dimension + elif facet == "edge": + dimension = self.edge_dimension + elif facet == "face": + dimension = self.face_dimension + else: + raise ValueError(f"Invalid facet: {facet}. Must be one of: node, edge.") return self._create_data_array(data, dimension)