Fix type hint

Deltares · Mar 13, 2024 · 3d354cd · 3d354cd
1 parent b0565a8
commit 3d354cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/ribasim/ribasim/geometry/edge.py b/python/ribasim/ribasim/geometry/edge.py
@@ -78,9 +78,9 @@ def get_where_edge_type(self, edge_type: str) -> NDArray[np.bool_]:
         assert self.df is not None
         return (self.df.edge_type == edge_type).to_numpy()
 
-    def plot(self, **kwargs) -> Axes:
+    def plot(self, **kwargs) -> Axes | None:
         if self.df is None:
-            return
+            return None
         kwargs = kwargs.copy()  # Avoid side-effects
         ax = kwargs.get("ax", None)
         color_flow = kwargs.pop("color_flow", None)