diff --git a/utils/Quality_Relabel.py b/utils/Quality_Relabel.py index 2e7f1f6..ea45b0f 100644 --- a/utils/Quality_Relabel.py +++ b/utils/Quality_Relabel.py @@ -360,6 +360,8 @@ class Quality_Relabel: for j in event_code: if channel == "SpO2": mask = self.spo2_event_label[SP * self.frequency:EP * self.frequency] == j + elif channel == "orgdata" or channel == "0.7lowpass_resp": + mask = self.bcg_event_label[SP * self.frequency:EP * self.frequency] == j else: mask = self.ecg_event_label[SP * self.frequency:EP * self.frequency] == j if event_show_under: