changes

Author: jpillai00

src/rbc/bids/longitudinal.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, TypedDict
 
 from rbc.bids import Suffix, TemplateSpace, bids_safe_label
 
@@ -108,6 +108,18 @@ def export_longitudinal_anat(aex: Bids, outputs: AnatomicalLongOutputs) -> None:
     )
 
 
+class LongitudinalFuncInputs(TypedDict):
+    """Resolved inputs for the longitudinal functional workflow."""
+
+    template: Path
+    anat_to_template_xfm: Path
+    bold_to_anat_itk: Path
+    sbref: Path
+    bold: Path
+    bold_mask: Path
+    regressor_files: dict[str, Path]
+
+
 def resolve_longitudinal_func(
     func_q: Bids,
     tpl_q: Bids,
@@ -116,7 +128,7 @@ def resolve_longitudinal_func(
     *,
     ses: str,
     regressors: Sequence[str],
-) -> dict[str, Path | None | dict[str, Path]]:
+) -> LongitudinalFuncInputs:
     """Resolve inputs for longitudinal functional processing.
 
     Args:

src/rbc/cli/longitudinal.py

@@ -46,7 +46,7 @@ def validate_namespace(cls, ns: argparse.Namespace) -> LongitudinalArgs:
 def main(args: LongitudinalArgs) -> int:
     """Main entrypoint of longitudinal workflow."""
     run(
-        input_dirs=list(args.input_dirs),
+        input_dirs=args.input_dirs,
         output_dir=args.output_dir,
         filters=Filters(
             participant_label=args.participant_label,

src/rbc/orchestration/longitudinal.py

@@ -17,6 +17,7 @@
     load_table,
 )
 from rbc.bids.longitudinal import (
+    LongitudinalFuncInputs,
     export_longitudinal_anat,
     export_longitudinal_func,
     resolve_longitudinal_anat,
@@ -91,7 +92,7 @@ def process_func(
     func_q = pipe_ctx.bids(datatype=Datatype.FUNC, entities=ents)
     tpl_q = pipe_ctx.bids(datatype=Datatype.ANAT).derive(ses="longitudinal")
 
-    resolved = resolve_longitudinal_func(
+    resolved: LongitudinalFuncInputs = resolve_longitudinal_func(
         func_q,
         tpl_q,
         func_df,
@@ -105,7 +106,7 @@ def process_func(
 
 
 def run(
-    input_dirs: list[Path],
+    input_dirs: Sequence[Path],
     output_dir: Path,
     *,
     filters: Filters,