Return instead of crash when trying to plot model without edges

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