Fixed use of non-target/decoding channel in the combined setting (#380)

neuromodulation · Nov 13, 2024 · 6a3b916 · 6a3b916
1 parent 206a455
commit 6a3b916
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py_neuromodulation/analysis/decode.py b/py_neuromodulation/analysis/decode.py
@@ -168,7 +168,7 @@ def set_data(self, features):
             self.feature_names = [
                 col
                 for col in self.features.columns
-                if not (("time" in col) or (self.label_name in col))
+                if any(col.startswith(used_ch) for used_ch in self.used_chs)
             ]
             self.data = np.nan_to_num(np.array(self.features[self.feature_names]))