Update glm_spectrum.py

Fix #249 - implementing the proposed changes
OHBA-analysis · Jan 9, 2024 · 9a21eea · 9a21eea
1 parent 5e7fe04
commit 9a21eea
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/osl/glm/glm_spectrum.py b/osl/glm/glm_spectrum.py
@@ -634,7 +634,10 @@ def plot_joint_spectrum_clusters(xvect, psd, clusters, info, ax=None, freqs='aut
         freqs[clu[2][0]] = 1
         finds = np.where(freqs)[0]
         if len(finds) == 1:
-            finds = np.array([finds[0], finds[0]+1])
+            if finds[0]<len(fx[0])-1: 
+                finds = np.array([finds[0], finds[0]+1])
+            else: # can't extend to next freq if last freq
+                finds = np.array([finds[0], finds[0]])
 
         msg = 'Cluster {} - stat: {}, freq range: {}, num channels {}'
         freq_range = (fx[0][finds[0]], fx[0][finds[-1]])