From b5d1e16234f0745c8407ea367f751a9a759787b1 Mon Sep 17 00:00:00 2001
From: David Keitel <david.keitel@ligo.org>
Date: Thu, 10 Sep 2020 09:26:39 +0200
Subject: [PATCH] allow single-param plot with np.atleast_2d(axes)
---
gridcorner.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/gridcorner.py b/gridcorner.py
index 304f0a2..0b71acb 100644
--- a/gridcorner.py
+++ b/gridcorner.py
@@ -56,6 +56,7 @@ def _get_fig_and_axes(ndim, factor, whspace):
plotdim = factor * ndim + factor * (ndim - 1.) * whspace
dim = lbdim + plotdim + trdim
fig, axes = plt.subplots(ndim, ndim, figsize=(dim, dim))
+ axes = np.atleast_2d(axes) # allow single-parameter plots
# Format the figure.
lb = lbdim / dim
--
GitLab