junifer 0.0.5.dev242__py3-none-any.whl → 0.0.6__py3-none-any.whl

junifer/markers/brainprint.py

@@ -8,10 +8,7 @@ from pathlib import Path
 from typing import (
     Any,
     ClassVar,
-    Dict,
-    List,
     Optional,
-    Set,
     Union,
 )
 
@@ -25,6 +22,7 @@ from ..external.BrainPrint.brainprint.brainprint import (
 )
 from ..external.BrainPrint.brainprint.surfaces import surf_to_vtk
 from ..pipeline import WorkDirManager
+from ..typing import Dependencies, ExternalDependencies, MarkerInOutMappings
 from ..utils import logger, run_ext_cmd
 from .base import BaseMarker
 
@@ -68,7 +66,7 @@ class BrainPrint(BaseMarker):
 
     """
 
-    _EXT_DEPENDENCIES: ClassVar[List[Dict[str, Union[str, List[str]]]]] = [
+    _EXT_DEPENDENCIES: ClassVar[ExternalDependencies] = [
         {
             "name": "freesurfer",
             "commands": [
@@ -80,9 +78,9 @@ class BrainPrint(BaseMarker):
         },
     ]
 
-    _DEPENDENCIES: ClassVar[Set[str]] = {"lapy", "numpy"}
+    _DEPENDENCIES: ClassVar[Dependencies] = {"lapy", "numpy"}
 
-    _MARKER_INOUT_MAPPINGS: ClassVar[Dict[str, Dict[str, str]]] = {
+    _MARKER_INOUT_MAPPINGS: ClassVar[MarkerInOutMappings] = {
         "FreeSurfer": {
             "eigenvalues": "scalar_table",
             "areas": "vector",
@@ -117,7 +115,7 @@ class BrainPrint(BaseMarker):
         self,
         aseg_path: Path,
         norm_path: Path,
-        indices: List,
+        indices: list,
     ) -> Path:
         """Generate a surface from the aseg and label files.
 
@@ -143,7 +141,7 @@ class BrainPrint(BaseMarker):
         mri_binarize_cmd = [
             "mri_binarize",
             f"--i {aseg_path.resolve()}",
-            f"--match {''.join(indices)}",
+            f"--match {' '.join(indices)}",
             f"--o {mri_binarize_output_path.resolve()}",
         ]
         # Call mri_binarize command
@@ -193,7 +191,7 @@ class BrainPrint(BaseMarker):
         self,
         aseg_path: Path,
         norm_path: Path,
-    ) -> Dict[str, Path]:
+    ) -> dict[str, Path]:
         """Create surfaces from FreeSurfer aseg labels.
 
         Parameters
@@ -268,7 +266,7 @@ class BrainPrint(BaseMarker):
         rh_white_path: Path,
         lh_pial_path: Path,
         rh_pial_path: Path,
-    ) -> Dict[str, Path]:
+    ) -> dict[str, Path]:
         """Create cortical surfaces from FreeSurfer labels.
 
         Parameters
@@ -309,7 +307,7 @@ class BrainPrint(BaseMarker):
 
     def _fix_nan(
         self,
-        input_data: List[Union[float, str, npt.ArrayLike]],
+        input_data: list[Union[float, str, npt.ArrayLike]],
     ) -> np.ndarray:
         """Convert BrainPrint output with string NaN to ``numpy.nan``.
 
@@ -330,9 +328,9 @@ class BrainPrint(BaseMarker):
 
     def compute(
         self,
-        input: Dict[str, Any],
-        extra_input: Optional[Dict] = None,
-    ) -> Dict:
+        input: dict[str, Any],
+        extra_input: Optional[dict] = None,
+    ) -> dict:
         """Compute.
 
         Parameters

junifer/markers/complexity/__init__.py

@@ -4,7 +4,6 @@
 #          Synchon Mandal <s.mandal@fz-juelich.de>
 # License: AGPL
 
-
 from importlib.util import find_spec
 
 from ..utils import raise_error
@@ -17,21 +16,6 @@ if find_spec("neurokit2") is None:
         klass=ImportError,
     )
 else:
-    # Import markers
-    from .hurst_exponent import HurstExponent
-    from .range_entropy import RangeEntropy
-    from .range_entropy_auc import RangeEntropyAUC
-    from .perm_entropy import PermEntropy
-    from .weighted_perm_entropy import WeightedPermEntropy
-    from .sample_entropy import SampleEntropy
-    from .multiscale_entropy_auc import MultiscaleEntropyAUC
+    import lazy_loader as lazy
 
-    __all__ = [
-        "HurstExponent",
-        "RangeEntropy",
-        "RangeEntropyAUC",
-        "PermEntropy",
-        "WeightedPermEntropy",
-        "SampleEntropy",
-        "MultiscaleEntropyAUC",
-    ]
+    __getattr__, __dir__, __all__ = lazy.attach_stub(__name__, __file__)

junifer/markers/complexity/__init__.pyi

@@ -0,0 +1,17 @@
+__all__ = [
+    "HurstExponent",
+    "RangeEntropy",
+    "RangeEntropyAUC",
+    "PermEntropy",
+    "WeightedPermEntropy",
+    "SampleEntropy",
+    "MultiscaleEntropyAUC",
+]
+
+from .hurst_exponent import HurstExponent
+from .range_entropy import RangeEntropy
+from .range_entropy_auc import RangeEntropyAUC
+from .perm_entropy import PermEntropy
+from .weighted_perm_entropy import WeightedPermEntropy
+from .sample_entropy import SampleEntropy
+from .multiscale_entropy_auc import MultiscaleEntropyAUC

junifer/markers/complexity/complexity_base.py

@@ -8,13 +8,11 @@ from typing import (
     TYPE_CHECKING,
     Any,
     ClassVar,
-    Dict,
-    List,
     Optional,
-    Set,
     Union,
 )
 
+from ...typing import Dependencies, MarkerInOutMappings
 from ...utils import raise_error
 from ..base import BaseMarker
 from ..parcel_aggregation import ParcelAggregation
@@ -51,9 +49,9 @@ class ComplexityBase(BaseMarker):
 
     """
 
-    _DEPENDENCIES: ClassVar[Set[str]] = {"nilearn", "neurokit2"}
+    _DEPENDENCIES: ClassVar[Dependencies] = {"nilearn", "neurokit2"}
 
-    _MARKER_INOUT_MAPPINGS: ClassVar[Dict[str, Dict[str, str]]] = {
+    _MARKER_INOUT_MAPPINGS: ClassVar[MarkerInOutMappings] = {
         "BOLD": {
             "complexity": "vector",
         },
@@ -61,10 +59,10 @@ class ComplexityBase(BaseMarker):
 
     def __init__(
         self,
-        parcellation: Union[str, List[str]],
+        parcellation: Union[str, list[str]],
         agg_method: str = "mean",
-        agg_method_params: Optional[Dict] = None,
-        masks: Union[str, Dict, List[Union[Dict, str]], None] = None,
+        agg_method_params: Optional[dict] = None,
+        masks: Union[str, dict, list[Union[dict, str]], None] = None,
         name: Optional[str] = None,
     ) -> None:
         self.parcellation = parcellation
@@ -86,9 +84,9 @@ class ComplexityBase(BaseMarker):
 
     def compute(
         self,
-        input: Dict[str, Any],
-        extra_input: Optional[Dict[str, Any]] = None,
-    ) -> Dict[str, Any]:
+        input: dict[str, Any],
+        extra_input: Optional[dict[str, Any]] = None,
+    ) -> dict[str, Any]:
         """Compute.
 
         Parameters

junifer/markers/complexity/hurst_exponent.py

@@ -4,7 +4,7 @@
 #          Leonard Sasse <l.sasse@fz-juelich.de>
 # License: AGPL
 
-from typing import Dict, List, Optional, Union
+from typing import Optional, Union
 
 import neurokit2 as nk
 import numpy as np
@@ -24,8 +24,8 @@ class HurstExponent(ComplexityBase):
     Parameters
     ----------
     parcellation : str or list of str
-        The name(s) of the parcellation(s). Check valid options by calling
-        :func:`junifer.data.parcellations.list_parcellations`.
+        The name(s) of the parcellation(s) to use.
+        See :func:`.list_data` for options.
     agg_method : str, optional
         The method to perform aggregation using. Check valid options in
         :func:`junifer.stats.get_aggfunc_by_name` (default "mean").
@@ -55,11 +55,11 @@ class HurstExponent(ComplexityBase):
 
     def __init__(
         self,
-        parcellation: Union[str, List[str]],
+        parcellation: Union[str, list[str]],
         agg_method: str = "mean",
-        agg_method_params: Optional[Dict] = None,
-        masks: Union[str, Dict, List[Union[Dict, str]], None] = None,
-        params: Optional[Dict] = None,
+        agg_method_params: Optional[dict] = None,
+        masks: Union[str, dict, list[Union[dict, str]], None] = None,
+        params: Optional[dict] = None,
         name: Optional[str] = None,
     ) -> None:
         super().__init__(

junifer/markers/complexity/multiscale_entropy_auc.py

@@ -4,7 +4,7 @@
 #          Leonard Sasse <l.sasse@fz-juelich.de>
 # License: AGPL
 
-from typing import Dict, List, Optional, Union
+from typing import Optional, Union
 
 import neurokit2 as nk
 import numpy as np
@@ -24,8 +24,8 @@ class MultiscaleEntropyAUC(ComplexityBase):
     Parameters
     ----------
     parcellation : str or list of str
-        The name(s) of the parcellation(s). Check valid options by calling
-        :func:`junifer.data.parcellations.list_parcellations`.
+        The name(s) of the parcellation(s) to use.
+        See :func:`.list_data` for options.
     agg_method : str, optional
         The method to perform aggregation using. Check valid options in
         :func:`junifer.stats.get_aggfunc_by_name` (default "mean").
@@ -56,11 +56,11 @@ class MultiscaleEntropyAUC(ComplexityBase):
 
     def __init__(
         self,
-        parcellation: Union[str, List[str]],
+        parcellation: Union[str, list[str]],
         agg_method: str = "mean",
-        agg_method_params: Optional[Dict] = None,
-        masks: Union[str, Dict, List[Union[Dict, str]], None] = None,
-        params: Optional[Dict] = None,
+        agg_method_params: Optional[dict] = None,
+        masks: Union[str, dict, list[Union[dict, str]], None] = None,
+        params: Optional[dict] = None,
         name: Optional[str] = None,
     ) -> None:
         super().__init__(

junifer/markers/complexity/perm_entropy.py

@@ -4,7 +4,7 @@
 #          Leonard Sasse <l.sasse@fz-juelich.de>
 # License: AGPL
 
-from typing import Dict, List, Optional, Union
+from typing import Optional, Union
 
 import neurokit2 as nk
 import numpy as np
@@ -24,8 +24,8 @@ class PermEntropy(ComplexityBase):
     Parameters
     ----------
     parcellation : str or list of str
-        The name(s) of the parcellation(s). Check valid options by calling
-        :func:`junifer.data.parcellations.list_parcellations`.
+        The name(s) of the parcellation(s) to use.
+        See :func:`.list_data` for options.
     agg_method : str, optional
         The method to perform aggregation using. Check valid options in
         :func:`junifer.stats.get_aggfunc_by_name` (default "mean").
@@ -56,11 +56,11 @@ class PermEntropy(ComplexityBase):
 
     def __init__(
         self,
-        parcellation: Union[str, List[str]],
+        parcellation: Union[str, list[str]],
         agg_method: str = "mean",
-        agg_method_params: Optional[Dict] = None,
-        masks: Union[str, Dict, List[Union[Dict, str]], None] = None,
-        params: Optional[Dict] = None,
+        agg_method_params: Optional[dict] = None,
+        masks: Union[str, dict, list[Union[dict, str]], None] = None,
+        params: Optional[dict] = None,
         name: Optional[str] = None,
     ) -> None:
         super().__init__(

junifer/markers/complexity/range_entropy.py

@@ -4,7 +4,7 @@
 #          Leonard Sasse <l.sasse@fz-juelich.de>
 # License: AGPL
 
-from typing import Dict, List, Optional, Union
+from typing import Optional, Union
 
 import neurokit2 as nk
 import numpy as np
@@ -24,8 +24,8 @@ class RangeEntropy(ComplexityBase):
     Parameters
     ----------
     parcellation : str or list of str
-        The name(s) of the parcellation(s). Check valid options by calling
-        :func:`junifer.data.parcellations.list_parcellations`.
+        The name(s) of the parcellation(s) to use.
+        See :func:`.list_data` for options.
     agg_method : str, optional
         The method to perform aggregation using. Check valid options in
         :func:`junifer.stats.get_aggfunc_by_name` (default "mean").
@@ -56,11 +56,11 @@ class RangeEntropy(ComplexityBase):
 
     def __init__(
         self,
-        parcellation: Union[str, List[str]],
+        parcellation: Union[str, list[str]],
         agg_method: str = "mean",
-        agg_method_params: Optional[Dict] = None,
-        masks: Union[str, Dict, List[Union[Dict, str]], None] = None,
-        params: Optional[Dict] = None,
+        agg_method_params: Optional[dict] = None,
+        masks: Union[str, dict, list[Union[dict, str]], None] = None,
+        params: Optional[dict] = None,
         name: Optional[str] = None,
     ) -> None:
         super().__init__(

junifer/markers/complexity/range_entropy_auc.py

@@ -4,7 +4,7 @@
 #          Leonard Sasse <l.sasse@fz-juelich.de>
 # License: AGPL
 
-from typing import Dict, List, Optional, Union
+from typing import Optional, Union
 
 import neurokit2 as nk
 import numpy as np
@@ -24,8 +24,8 @@ class RangeEntropyAUC(ComplexityBase):
     Parameters
     ----------
     parcellation : str or list of str
-        The name(s) of the parcellation(s). Check valid options by calling
-        :func:`junifer.data.parcellations.list_parcellations`.
+        The name(s) of the parcellation(s) to use.
+        See :func:`.list_data` for options.
     agg_method : str, optional
         The method to perform aggregation using. Check valid options in
         :func:`junifer.stats.get_aggfunc_by_name` (default "mean").
@@ -56,11 +56,11 @@ class RangeEntropyAUC(ComplexityBase):
 
     def __init__(
         self,
-        parcellation: Union[str, List[str]],
+        parcellation: Union[str, list[str]],
         agg_method: str = "mean",
-        agg_method_params: Optional[Dict] = None,
-        masks: Union[str, Dict, List[Union[Dict, str]], None] = None,
-        params: Optional[Dict] = None,
+        agg_method_params: Optional[dict] = None,
+        masks: Union[str, dict, list[Union[dict, str]], None] = None,
+        params: Optional[dict] = None,
         name: Optional[str] = None,
     ) -> None:
         super().__init__(

junifer/markers/complexity/sample_entropy.py

@@ -4,7 +4,7 @@
 #          Leonard Sasse <l.sasse@fz-juelich.de>
 # License: AGPL
 
-from typing import Dict, List, Optional, Union
+from typing import Optional, Union
 
 import neurokit2 as nk
 import numpy as np
@@ -24,8 +24,8 @@ class SampleEntropy(ComplexityBase):
     Parameters
     ----------
     parcellation : str or list of str
-        The name(s) of the parcellation(s). Check valid options by calling
-        :func:`junifer.data.parcellations.list_parcellations`.
+        The name(s) of the parcellation(s) to use.
+        See :func:`.list_data` for options.
     agg_method : str, optional
         The method to perform aggregation using. Check valid options in
         :func:`junifer.stats.get_aggfunc_by_name` (default "mean").
@@ -57,11 +57,11 @@ class SampleEntropy(ComplexityBase):
 
     def __init__(
         self,
-        parcellation: Union[str, List[str]],
+        parcellation: Union[str, list[str]],
         agg_method: str = "mean",
-        agg_method_params: Optional[Dict] = None,
-        masks: Union[str, Dict, List[Union[Dict, str]], None] = None,
-        params: Optional[Dict] = None,
+        agg_method_params: Optional[dict] = None,
+        masks: Union[str, dict, list[Union[dict, str]], None] = None,
+        params: Optional[dict] = None,
         name: Optional[str] = None,
     ) -> None:
         super().__init__(

junifer/markers/complexity/tests/test_complexity_base.py

@@ -8,7 +8,7 @@ import pytest
 
 pytest.importorskip("neurokit2")
 
-from junifer.markers.complexity.complexity_base import (  # noqa: E402
+from junifer.markers.complexity.complexity_base import (
     ComplexityBase,
 )
 

junifer/markers/complexity/tests/test_hurst_exponent.py

@@ -12,11 +12,11 @@ import pytest
 pytest.importorskip("neurokit2")
 
 
-from junifer.datareader import DefaultDataReader  # noqa: E402
-from junifer.markers.complexity import HurstExponent  # noqa: E402
-from junifer.pipeline.utils import _check_ants  # noqa: E402
-from junifer.storage import SQLiteFeatureStorage  # noqa: E402
-from junifer.testing.datagrabbers import (  # noqa: E402
+from junifer.datareader import DefaultDataReader
+from junifer.markers.complexity import HurstExponent
+from junifer.pipeline.utils import _check_ants
+from junifer.storage import SQLiteFeatureStorage
+from junifer.testing.datagrabbers import (
     SPMAuditoryTestingDataGrabber,
 )
 

junifer/markers/complexity/tests/test_multiscale_entropy_auc.py

@@ -11,11 +11,11 @@ import pytest
 
 pytest.importorskip("neurokit2")
 
-from junifer.datareader import DefaultDataReader  # noqa: E402
-from junifer.markers.complexity import MultiscaleEntropyAUC  # noqa: E402
-from junifer.pipeline.utils import _check_ants  # noqa: E402
-from junifer.storage import SQLiteFeatureStorage  # noqa: E402
-from junifer.testing.datagrabbers import (  # noqa: E402
+from junifer.datareader import DefaultDataReader
+from junifer.markers.complexity import MultiscaleEntropyAUC
+from junifer.pipeline.utils import _check_ants
+from junifer.storage import SQLiteFeatureStorage
+from junifer.testing.datagrabbers import (
     SPMAuditoryTestingDataGrabber,
 )
 

junifer/markers/complexity/tests/test_perm_entropy.py

@@ -11,11 +11,11 @@ import pytest
 
 pytest.importorskip("neurokit2")
 
-from junifer.datareader import DefaultDataReader  # noqa: E402
-from junifer.markers.complexity import PermEntropy  # noqa: E402
-from junifer.pipeline.utils import _check_ants  # noqa: E402
-from junifer.storage import SQLiteFeatureStorage  # noqa: E402
-from junifer.testing.datagrabbers import (  # noqa: E402
+from junifer.datareader import DefaultDataReader
+from junifer.markers.complexity import PermEntropy
+from junifer.pipeline.utils import _check_ants
+from junifer.storage import SQLiteFeatureStorage
+from junifer.testing.datagrabbers import (
     SPMAuditoryTestingDataGrabber,
 )
 

junifer/markers/complexity/tests/test_range_entropy.py

@@ -12,11 +12,11 @@ import pytest
 pytest.importorskip("neurokit2")
 
 
-from junifer.datareader import DefaultDataReader  # noqa: E402
-from junifer.markers.complexity import RangeEntropy  # noqa: E402
-from junifer.pipeline.utils import _check_ants  # noqa: E402
-from junifer.storage import SQLiteFeatureStorage  # noqa: E402
-from junifer.testing.datagrabbers import (  # noqa: E402
+from junifer.datareader import DefaultDataReader
+from junifer.markers.complexity import RangeEntropy
+from junifer.pipeline.utils import _check_ants
+from junifer.storage import SQLiteFeatureStorage
+from junifer.testing.datagrabbers import (
     SPMAuditoryTestingDataGrabber,
 )
 

junifer/markers/complexity/tests/test_range_entropy_auc.py

@@ -12,11 +12,11 @@ import pytest
 pytest.importorskip("neurokit2")
 
 
-from junifer.datareader import DefaultDataReader  # noqa: E402
-from junifer.markers.complexity import RangeEntropyAUC  # noqa: E402
-from junifer.pipeline.utils import _check_ants  # noqa: E402
-from junifer.storage import SQLiteFeatureStorage  # noqa: E402
-from junifer.testing.datagrabbers import (  # noqa: E402
+from junifer.datareader import DefaultDataReader
+from junifer.markers.complexity import RangeEntropyAUC
+from junifer.pipeline.utils import _check_ants
+from junifer.storage import SQLiteFeatureStorage
+from junifer.testing.datagrabbers import (
     SPMAuditoryTestingDataGrabber,
 )
 

junifer/markers/complexity/tests/test_sample_entropy.py

@@ -11,11 +11,11 @@ import pytest
 
 pytest.importorskip("neurokit2")
 
-from junifer.datareader import DefaultDataReader  # noqa: E402
-from junifer.markers.complexity import SampleEntropy  # noqa: E402
-from junifer.pipeline.utils import _check_ants  # noqa: E402
-from junifer.storage import SQLiteFeatureStorage  # noqa: E402
-from junifer.testing.datagrabbers import (  # noqa: E402
+from junifer.datareader import DefaultDataReader
+from junifer.markers.complexity import SampleEntropy
+from junifer.pipeline.utils import _check_ants
+from junifer.storage import SQLiteFeatureStorage
+from junifer.testing.datagrabbers import (
     SPMAuditoryTestingDataGrabber,
 )
 

junifer/markers/complexity/tests/test_weighted_perm_entropy.py

@@ -11,11 +11,11 @@ import pytest
 
 pytest.importorskip("neurokit2")
 
-from junifer.datareader import DefaultDataReader  # noqa: E402
-from junifer.markers.complexity import WeightedPermEntropy  # noqa: E402
-from junifer.pipeline.utils import _check_ants  # noqa: E402
-from junifer.storage import SQLiteFeatureStorage  # noqa: E402
-from junifer.testing.datagrabbers import (  # noqa: E402
+from junifer.datareader import DefaultDataReader
+from junifer.markers.complexity import WeightedPermEntropy
+from junifer.pipeline.utils import _check_ants
+from junifer.storage import SQLiteFeatureStorage
+from junifer.testing.datagrabbers import (
     SPMAuditoryTestingDataGrabber,
 )
 

junifer/markers/complexity/weighted_perm_entropy.py

@@ -4,7 +4,7 @@
 #          Leonard Sasse <l.sasse@fz-juelich.de>
 # License: AGPL
 
-from typing import Dict, List, Optional, Union
+from typing import Optional, Union
 
 import neurokit2 as nk
 import numpy as np
@@ -24,8 +24,8 @@ class WeightedPermEntropy(ComplexityBase):
     Parameters
     ----------
     parcellation : str or list of str
-        The name(s) of the parcellation(s). Check valid options by calling
-        :func:`junifer.data.parcellations.list_parcellations`.
+        The name(s) of the parcellation(s) to use.
+        See :func:`.list_data` for options.
     agg_method : str, optional
         The method to perform aggregation using. Check valid options in
         :func:`junifer.stats.get_aggfunc_by_name` (default "mean").
@@ -57,11 +57,11 @@ class WeightedPermEntropy(ComplexityBase):
 
     def __init__(
         self,
-        parcellation: Union[str, List[str]],
+        parcellation: Union[str, list[str]],
         agg_method: str = "mean",
-        agg_method_params: Optional[Dict] = None,
-        masks: Union[str, Dict, List[Union[Dict, str]], None] = None,
-        params: Optional[Dict] = None,
+        agg_method_params: Optional[dict] = None,
+        masks: Union[str, dict, list[Union[dict, str]], None] = None,
+        params: Optional[dict] = None,
         name: Optional[str] = None,
     ) -> None:
         super().__init__(

junifer/markers/ets_rss.py

@@ -6,11 +6,12 @@
 #          Synchon Mandal <s.mandal@fz-juelich.de>
 # License: AGPL
 
-from typing import Any, ClassVar, Dict, List, Optional, Set, Union
+from typing import Any, ClassVar, Optional, Union
 
 import numpy as np
 
 from ..api.decorators import register_marker
+from ..typing import Dependencies, MarkerInOutMappings
 from ..utils import logger
 from .base import BaseMarker
 from .parcel_aggregation import ParcelAggregation
@@ -27,8 +28,8 @@ class RSSETSMarker(BaseMarker):
     Parameters
     ----------
     parcellation : str or list of str
-        The name(s) of the parcellation(s). Check valid options by calling
-        :func:`.list_parcellations`.
+        The name(s) of the parcellation(s) to use.
+        See :func:`.list_data` for options.
     agg_method : str, optional
         The method to perform aggregation using. Check valid options in
         :func:`.get_aggfunc_by_name` (default "mean").
@@ -45,9 +46,9 @@ class RSSETSMarker(BaseMarker):
 
     """
 
-    _DEPENDENCIES: ClassVar[Set[str]] = {"nilearn"}
+    _DEPENDENCIES: ClassVar[Dependencies] = {"nilearn"}
 
-    _MARKER_INOUT_MAPPINGS: ClassVar[Dict[str, Dict[str, str]]] = {
+    _MARKER_INOUT_MAPPINGS: ClassVar[MarkerInOutMappings] = {
         "BOLD": {
             "rss_ets": "timeseries",
         },
@@ -55,10 +56,10 @@ class RSSETSMarker(BaseMarker):
 
     def __init__(
         self,
-        parcellation: Union[str, List[str]],
+        parcellation: Union[str, list[str]],
         agg_method: str = "mean",
-        agg_method_params: Optional[Dict] = None,
-        masks: Union[str, Dict, List[Union[Dict, str]], None] = None,
+        agg_method_params: Optional[dict] = None,
+        masks: Union[str, dict, list[Union[dict, str]], None] = None,
         name: Optional[str] = None,
     ) -> None:
         self.parcellation = parcellation
@@ -69,9 +70,9 @@ class RSSETSMarker(BaseMarker):
 
     def compute(
         self,
-        input: Dict[str, Any],
-        extra_input: Optional[Dict] = None,
-    ) -> Dict:
+        input: dict[str, Any],
+        extra_input: Optional[dict] = None,
+    ) -> dict:
         """Compute.
 
         Take a timeseries of brain areas, and calculate timeseries for each

junifer/markers/falff/__init__.py

@@ -3,8 +3,7 @@
 # Authors: Federico Raimondo <f.raimondo@fz-juelich.de>
 # License: AGPL
 
-from .falff_parcels import ALFFParcels
-from .falff_spheres import ALFFSpheres
+import lazy_loader as lazy
 
 
-__all__ = ["ALFFParcels", "ALFFSpheres"]
+__getattr__, __dir__, __all__ = lazy.attach_stub(__name__, __file__)