diff --git a/TPTBox/core/bids_files.py b/TPTBox/core/bids_files.py index c3b1423..44b536a 100755 --- a/TPTBox/core/bids_files.py +++ b/TPTBox/core/bids_files.py @@ -1987,6 +1987,7 @@ def loop_dict( key_transform (typing.Callable[[BIDS_FILE], str | None]): provide alternative dict name for certain fils, if default should be used return None key_addendum (list[str] | None, optional): Extra info-key names appended to each family's dict keys to disambiguate otherwise-identical entries. Defaults to None. + Returns: typing.Iterator[typing.Dict[str, BIDS_FILE | list[BIDS_FILE]]] """ diff --git a/TPTBox/core/internal/train_nnUnet/_prep_ds.py b/TPTBox/core/internal/train_nnUnet/_prep_ds.py index 64f5e51..637a692 100644 --- a/TPTBox/core/internal/train_nnUnet/_prep_ds.py +++ b/TPTBox/core/internal/train_nnUnet/_prep_ds.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import hashlib import json import random diff --git a/TPTBox/core/internal/train_nnUnet/fastProcessor.py b/TPTBox/core/internal/train_nnUnet/fastProcessor.py index 7417593..0e08ca1 100644 --- a/TPTBox/core/internal/train_nnUnet/fastProcessor.py +++ b/TPTBox/core/internal/train_nnUnet/fastProcessor.py @@ -12,6 +12,8 @@ # See the License for the specific language governing permissions and # limitations under the License. +from __future__ import annotations + import math from copy import deepcopy from pathlib import Path diff --git a/TPTBox/mesh3D/html_preview.py b/TPTBox/mesh3D/html_preview.py index 0012334..5ccd416 100644 --- a/TPTBox/mesh3D/html_preview.py +++ b/TPTBox/mesh3D/html_preview.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from dataclasses import dataclass from pathlib import Path from typing import Literal diff --git a/TPTBox/segmentation/VibeSeg/vibeseg.py b/TPTBox/segmentation/VibeSeg/vibeseg.py index 49222ff..4268ed7 100644 --- a/TPTBox/segmentation/VibeSeg/vibeseg.py +++ b/TPTBox/segmentation/VibeSeg/vibeseg.py @@ -127,7 +127,7 @@ def run_vibeseg( args[k] = v return run_inference_on_file( dataset_id, - [to_nii(i)] if not isinstance(i, list | tuple) else [to_nii(j) for j in i], + [to_nii(i)] if not isinstance(i, (list, tuple)) else [to_nii(j) for j in i], out_file=out_seg, override=override, gpu=gpu, diff --git a/TPTBox/segmentation/_deface.py b/TPTBox/segmentation/_deface.py index 84d507f..7e3f959 100644 --- a/TPTBox/segmentation/_deface.py +++ b/TPTBox/segmentation/_deface.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from pathlib import Path import numpy as np