Fix mode selection in analyze_variation

Addressing issue zlatko-minev#148: analyze_variation incorrectly chooses Pj, Sj, Om, PHI_zpf when passed a subset of modes
clarayfontaine · Apr 4, 2023 · 6d323b0 · 6d323b0
1 parent 017ea78
commit 6d323b0
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/pyEPR/core_quantum_analysis.py b/pyEPR/core_quantum_analysis.py
@@ -664,11 +664,11 @@ def analyze_variation(self,
             PJ_cap = PJ_cap[:, junctions]
 
         if modes is not None:
-            freqs_hfss = freqs_hfss[range(len(self.modes[variation])), ]
-            PJ = PJ[range(len(modes)), :]
-            SJ = SJ[range(len(modes)), :]
-            Om = Om[range(len(modes)), :][:, range(len(modes))]
-            PHI_zpf = PHI_zpf[range(len(modes)), :]
+            freqs_hfss = freqs_hfss[variation].values[(modes)]
+            PJ = PJ[modes, :]
+            SJ = SJ[modes, :]
+            Om = Om[modes, :][:, modes]
+            PHI_zpf = PHI_zpf[modes, :]
             PJ_cap = PJ_cap[:, junctions]
 
         # Analytic 4-th order