From a4ccc21a9e21fd3bbd9cde4f3743fee4ecac59b6 Mon Sep 17 00:00:00 2001 From: mathiasg Date: Mon, 14 Oct 2024 16:49:16 -0400 Subject: [PATCH] DOC/STY: Fix doctest, include type hint --- nibabies/workflows/base.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/nibabies/workflows/base.py b/nibabies/workflows/base.py index a6800634..eca233a3 100644 --- a/nibabies/workflows/base.py +++ b/nibabies/workflows/base.py @@ -70,11 +70,13 @@ from bids.layout import BIDSLayout from niworkflows.utils.spaces import SpatialReferences + SubjectSession = tuple[str, str | None] + AUTO_T2W_MAX_AGE = 8 -def init_nibabies_wf(subworkflows_list): +def init_nibabies_wf(subworkflows_list: list[SubjectSession]): """ Build *NiBabies*'s pipeline. @@ -92,7 +94,7 @@ def init_nibabies_wf(subworkflows_list): from nibabies.workflows.tests import mock_config from nibabies.workflows.base import init_nibabies_wf with mock_config(): - wf = init_nibabies_wf() + wf = init_nibabies_wf(['01', None]) Parameters ----------