fix bug psd name writing

neuromodulation · Nov 11, 2024 · 8cf5156 · 8cf5156
1 parent 647a0a1
commit 8cf5156
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py_neuromodulation/features/oscillatory.py b/py_neuromodulation/features/oscillatory.py
@@ -176,7 +176,7 @@ def calc_feature(self, data: np.ndarray) -> dict:
         if self.settings.return_spectrum:
             combinations = product(enumerate(self.ch_names), enumerate(self.freqs))
             for (ch_idx, ch_name), (idx, f) in combinations:
-                feature_results[f"{ch_name}_welch_psd_{str(f)}"] = Z[ch_idx][idx]
+                feature_results[f"{ch_name}_welch_psd_{int(f)}"] = Z[ch_idx][idx]
 
         return feature_results
 
@@ -242,7 +242,7 @@ def calc_feature(self, data: np.ndarray) -> dict:
         if self.settings.return_spectrum:
             combinations = product(enumerate(self.ch_names), enumerate(self.freqs))
             for (ch_idx, ch_name), (idx, f) in combinations:
-                feature_results[f"{ch_name}_stft_psd_{str(f)}"] = Z[ch_idx].mean(
+                feature_results[f"{ch_name}_stft_psd_{int(f)}"] = Z[ch_idx].mean(
                     axis=1
                 )[idx]