fix apply_estimator sw

neuromodulation · Feb 11, 2025 · d263661 · d263661
1 parent aab97fd
commit d263661
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/py_neuromodulation/features/sharpwaves.py b/py_neuromodulation/features/sharpwaves.py
@@ -302,9 +302,11 @@ def calc_feature(self, data: np.ndarray) -> dict:
                 )
             # add here also the num_peaks features
             if self.sw_settings.sharpwave_features.num_peaks:
-                key_name = f"{ch_name}_Sharpwave_num_peaks_{filter_name}"
-                feature_results[key_name] = np_mean([dict_ch_features[key_name]["Peak"],
-                                                    dict_ch_features[key_name]["Trough"]])
+                for ch_name in self.ch_names:
+                    for filter_name in self.filter_names:
+                        key_name = f"{ch_name}_Sharpwave_num_peaks_{filter_name}"
+                        feature_results[key_name] = np_mean([dict_ch_features[key_name]["Peak"],
+                                                            dict_ch_features[key_name]["Trough"]])
         else:
             # otherwise, save all write all "flattened" key value pairs in feature_results
             for key, subdict in dict_ch_features.items():