Update run_herg_qc.py

Allow for protocols in export_config.py that are not included in the dataset
CardiacModelling · Dec 11, 2024 · 7b86e2e · 7b86e2e
1 parent cdbc4ed
commit 7b86e2e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pcpostprocess/scripts/run_herg_qc.py b/pcpostprocess/scripts/run_herg_qc.py
@@ -95,6 +95,8 @@ def main():
     sys.modules['export_config'] = export_config
     spec.loader.exec_module(export_config)
 
+    data_list = os.listdir(args.data_directory)
+    export_config.D2S_QC =  {x:y for x,y in export_config.D2S_QC.items() if any([x == '_'.join(z.split('_')[:-1]) for z in data_list])}
     export_config.savedir = args.output_dir
 
     args.saveID = export_config.saveID