FIX recognise acq; this is a hot fix

SIMEXP · Dec 8, 2023 · ee6f80f · ee6f80f
1 parent 5d5a16b
commit ee6f80f
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/giga_auto_qc/tests/test_utils.py b/giga_auto_qc/tests/test_utils.py
@@ -45,3 +45,16 @@ def test_parse_scan_information():
     )
     parsed = utils.parse_scan_information(metrics=metrics)
     assert list(parsed.columns[:3]) == ["participant_id", "task", "run"]
+
+    # specifiers with different entities
+    bids_specifier_index = [
+        "sub-test_task-finger",
+        "sub-test_task-rest_acq-1_run-001",
+        "sub-test_task-rest_acq-2_run-001",
+        "sub-test_task-rest_run-002",
+    ]
+    metrics = pd.DataFrame(
+        np.random.random((4, 4)), index=bids_specifier_index
+    )
+    parsed = utils.parse_scan_information(metrics=metrics)
+    assert list(parsed.columns[:4]) == ["participant_id", "task", "acq", "run"]
diff --git a/giga_auto_qc/utils.py b/giga_auto_qc/utils.py
@@ -2,7 +2,13 @@
 from pathlib import Path
 import pandas as pd
 
-BIDS_ENTITIES = {"sub": 0, "ses": 1, "task": 2, "run": 3}
+BIDS_ENTITIES = {
+    "sub": 0,
+    "ses": 1,
+    "task": 2,
+    "acq": 3,
+    "run": 4,
+}
 
 
 def get_subject_lists(