Merge pull request #375 from mj-will/fix-373-corner-plot-bug

BUG: fix 373 corner plot bug
mj-will · Mar 11, 2024 · b160e31 · b160e31
2 parents a9c912f + d6422a3
commit b160e31
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nessai/plot.py b/nessai/plot.py
@@ -636,7 +636,7 @@ def corner_plot(
     if len(labels) != unstruct_array.shape[-1]:
         labels = labels[has_range]
 
-    if truths:
+    if truths is not None:
         if isinstance(truths, dict):
             if include:
                 truths = np.array([truths[n] for n in include])

diff --git a/tests/test_plot.py b/tests/test_plot.py
@@ -503,7 +503,7 @@ def test_corner_plot_w_labels(live_points, labels):
 
 @pytest.mark.parametrize(
     "truths",
-    [[0, 0], [0, 0, None, None, None], {"x": 0, "y": 0}],
+    [[0, 0], [0, 0, None, None, None], {"x": 0, "y": 0}, np.array([0, 0])],
 )
 def test_corner_plot_w_truths(live_points, truths):
     """Test the corner plot with truths"""