Fix ConnectPlot (PennLINC#1188)

tsalo · Jun 27, 2024 · 14930fe · 14930fe
1 parent 921278b
commit 14930fe
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/xcp_d/interfaces/connectivity.py b/xcp_d/interfaces/connectivity.py
@@ -438,6 +438,9 @@ def _run_interface(self, runtime):
             ax_idx = list(range(ncols))
 
         fig, axes = plt.subplots(nrows=nrows, ncols=ncols, figsize=figsize)
+        if isinstance(axes, plt.Axes):
+            axes = np.array([axes])
+
         for i_ax, atlas in enumerate(selected_atlases):
             atlas_idx = self.inputs.atlases.index(atlas)
             atlas_file = self.inputs.correlations_tsv[atlas_idx]