fixed bug when theres only 1 axes

Jhsmit · Feb 4, 2020 · 64136a2 · 64136a2
1 parent a1aa2e2
commit 64136a2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyhdx/plot.py b/pyhdx/plot.py
@@ -85,6 +85,7 @@ def make_coverage_figure(pm, wrap, aa_per_subplot, figsize=(10, 8), **kwargs):
     num_axes = pm.stop // aa_per_subplot + 1
 
     fig, axes = plt.subplots(num_axes, figsize=figsize)
+    axes = [axes] if num_axes == 1 else axes
     for j, ax in enumerate(axes):
         i = -1