changes

coganlab · Dec 21, 2023 · cd1b6b9 · cd1b6b9
1 parent dad063b
commit cd1b6b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ieeg/viz/mri.py b/ieeg/viz/mri.py
@@ -563,7 +563,7 @@ def _add_electrodes(fig: mne.viz.Brain, info: mne.Info, hemi: str,
     n_groups = len(set(groups.values()))
     if colors is None:
         colors = parula.mat_colors[:n_groups]
-    elif not isinstance(colors, Iterable) or isinstance(colors, tuple):
+    elif not np.isscalar(colors) or isinstance(colors, tuple):
         colors = [colors] * n_groups
     else:
         colors = list(colors)