From c7b951ff0add426f89a89418661b38ad8d4fdf81 Mon Sep 17 00:00:00 2001 From: Hao-Ting Wang Date: Fri, 8 Dec 2023 16:13:53 -0500 Subject: [PATCH] FIX recognise acq; this is a hot fix (#31) --- giga_auto_qc/tests/test_utils.py | 13 +++++++++++++ giga_auto_qc/utils.py | 8 +++++++- 2 files changed, 20 insertions(+), 1 deletion(-) diff --git a/giga_auto_qc/tests/test_utils.py b/giga_auto_qc/tests/test_utils.py index 7d4c903..801ad85 100644 --- 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 index a7b2340..8665c50 100644 --- 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(