support linear trends in modelcompare

pastas · Feb 17, 2023 · 205faa9 · 205faa9
1 parent 3eac98a
commit 205faa9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pastas/modelcompare.py b/pastas/modelcompare.py
@@ -512,6 +512,8 @@ def plot_response(
                         continue
                     if response == "step":
                         step = ml.get_step_response(smn, add_0=True)
+                        if step is None:
+                            continue
                         self.axes[axn.format(i=j)].plot(
                             step.index,
                             step.values,
@@ -520,6 +522,8 @@ def plot_response(
                         )
                     elif response == "block":
                         block = ml.get_block_response(smn)
+                        if block is None:
+                            continue
                         self.axes[axn.format(i=j)].semilogx(
                             block.index,
                             block.values,