Merge branch 'python3' of https://github.com/nasa/SMCPy into python3

nasa · Oct 22, 2020 · fced850 · fced850
2 parents 0279ca6 + 92b751f
commit fced850
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/smcpy/utils/plotter.py b/smcpy/utils/plotter.py
@@ -89,10 +89,9 @@ def wrapper(self, *args, **kwargs):
 
 def plot_mcmc_chain(chain, param_labels, burnin=0, save=False, show=True,
                     include_kde=False, prefix='mcmc_chain'):
-    n_columns = 1
-    gridspec = None
+    n_columns = 2
+    gridspec = {'width_ratios': [1., 0.], 'wspace': 0.0}
     if include_kde:
-        n_columns += 1
         gridspec = {'width_ratios': [0.85, 0.15], 'wspace': 0.0}
 
     fig, ax = plt.subplots(len(param_labels), n_columns, sharey='row',
@@ -111,9 +110,9 @@ def plot_mcmc_chain(chain, param_labels, burnin=0, save=False, show=True,
                 ax[i, 1].fill_betweenx(x, kde.pdf(x), np.zeros(x.shape),
                                       alpha=0.3)
 
+        ax[i, 1].axis('off')
         if include_kde:
             ax[i, 1].set_xlim(0, None)
-            ax[i, 1].axis('off')
 
 
         ax[i, 0].set_ylabel(name)