From bad5db426d9a231a68f1737feb9117be471ddec0 Mon Sep 17 00:00:00 2001
From: Zhongweiyang Xu <42936522+XZWY@users.noreply.github.com>
Date: Sat, 22 Jul 2023 20:43:57 -0700
Subject: [PATCH] Update music.py

fix plot_individual_spectrum for music doa estimation
---
 pyroomacoustics/doa/music.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyroomacoustics/doa/music.py b/pyroomacoustics/doa/music.py
index 8bb85760..d0b5cf0b 100644
--- a/pyroomacoustics/doa/music.py
+++ b/pyroomacoustics/doa/music.py
@@ -118,14 +118,14 @@ def plot_individual_spectrum(self):
             import warnings
 
             warnings.warn("Only for 2D.")
-            return
+            # return
 
         # plot
         for k in range(self.num_freq):
             freq = float(self.freq_bins[k]) / self.nfft * self.fs
             azimuth = self.grid.azimuth * 180 / np.pi
 
-            plt.plot(azimuth, self.Pssl[k, 0 : len(azimuth)])
+            plt.plot(azimuth, self.Pssl[0 : len(azimuth), k])
 
             plt.ylabel("Magnitude")
             plt.xlabel("Azimuth [degrees]")