junifer 0.0.6.dev248__py3-none-any.whl → 0.0.6.dev252__py3-none-any.whl

junifer/external/nilearn/junifer_nifti_spheres_masker.py

@@ -3,7 +3,7 @@
 # Authors: Synchon Mandal <s.mandal@fz-juelich.de>
 # License: AGPL
 
-from typing import TYPE_CHECKING, Callable, List, Optional, Tuple, Union
+from typing import TYPE_CHECKING, Callable, Optional, Union
 
 import numpy as np
 from nilearn import image, masking
@@ -271,7 +271,7 @@ class _JuniferExtractionFunctor:
     def __call__(
         self,
         imgs: Union["Nifti1Image", "Nifti2Image"],
-    ) -> Tuple["ArrayLike", None]:
+    ) -> tuple["ArrayLike", None]:
         """Implement function call overloading.
 
         Parameters
@@ -372,9 +372,9 @@ class JuniferNiftiSpheresMasker(NiftiSpheresMasker):
         self,
         imgs: Union["Nifti1Image", "Nifti2Image"],
         confounds: Union[
-            str, "Path", "ArrayLike", "DataFrame", List, None
+            str, "Path", "ArrayLike", "DataFrame", list, None
         ] = None,
-        sample_mask: Union["ArrayLike", List, Tuple, None] = None,
+        sample_mask: Union["ArrayLike", list, tuple, None] = None,
     ) -> "ArrayLike":
         """Extract signals from a single 4D niimg.
 

junifer/external/nilearn/tests/test_junifer_connectivity_measure.py

@@ -6,7 +6,7 @@
 import copy
 import warnings
 from math import cosh, exp, log, sinh, sqrt
-from typing import TYPE_CHECKING, List, Optional, Tuple, Type, Union
+from typing import TYPE_CHECKING, Optional, Union
 
 import numpy as np
 import pytest
@@ -149,7 +149,7 @@ def random_spd(
 
 def _signals(
     n_subjects: int = N_SUBJECTS,
-) -> Tuple[List[np.ndarray], np.ndarray]:
+) -> tuple[list[np.ndarray], np.ndarray]:
     """Generate signals and compute covariances while applying confounds.
 
     Parameters
@@ -178,7 +178,7 @@ def _signals(
 
 
 @pytest.fixture
-def signals() -> List[np.ndarray]:
+def signals() -> list[np.ndarray]:
     """Return signals as list of np.ndarray."""
     return _signals(N_SUBJECTS)[0]
 
@@ -186,7 +186,7 @@ def signals() -> List[np.ndarray]:
 @pytest.fixture
 def signals_and_covariances(
     cov_estimator: Union[LedoitWolf, EmpiricalCovariance]
-) -> Tuple[List[np.ndarray], List[float]]:
+) -> tuple[list[np.ndarray], list[float]]:
     """Return signals and covariances for a covariance estimator.
 
     Parameters
@@ -417,7 +417,7 @@ def grad_geometric_mean(
     init: Optional["ArrayLike"] = None,
     max_iter: int = 10,
     tol: float = 1e-7,
-) -> List[float]:
+) -> list[float]:
     """Compute gradient of geometric mean.
 
     Return the norm of the covariant derivative at each iteration step
@@ -615,8 +615,8 @@ def test_connectivity_measure_errors():
 @pytest.mark.parametrize("kind", CONNECTIVITY_KINDS)
 def test_connectivity_measure_generic(
     kind: str,
-    cov_estimator: Type["BaseEstimator"],
-    signals_and_covariances: Tuple[List[np.ndarray], List[float]],
+    cov_estimator: type["BaseEstimator"],
+    signals_and_covariances: tuple[list[np.ndarray], list[float]],
 ) -> None:
     """Test generic JuniferConnectivityMeasure.
 
@@ -778,8 +778,8 @@ def _assert_connectivity_partial_correlation(connectivities, covs) -> None:
 )
 def test_connectivity_measure_specific_for_each_kind(
     kind: str,
-    cov_estimator: Type["BaseEstimator"],
-    signals_and_covariances: Tuple[List[np.ndarray], List[float]],
+    cov_estimator: type["BaseEstimator"],
+    signals_and_covariances: tuple[list[np.ndarray], list[float]],
 ) -> None:
     """Test connectivity matrix for each kind.
 
@@ -812,7 +812,7 @@ def test_connectivity_measure_specific_for_each_kind(
 
 @pytest.mark.parametrize("kind", CONNECTIVITY_KINDS)
 def test_connectivity_measure_check_mean(
-    kind: str, signals: List[np.ndarray]
+    kind: str, signals: list[np.ndarray]
 ) -> None:
     """Test mean of connectivity matrix for each kind.
 
@@ -846,7 +846,7 @@ def test_connectivity_measure_check_mean(
 
 @pytest.mark.parametrize("kind", CONNECTIVITY_KINDS)
 def test_connectivity_measure_check_vectorization_option(
-    kind: str, signals: List[np.ndarray]
+    kind: str, signals: list[np.ndarray]
 ) -> None:
     """Test vectorization of connectivity matrix for each kind.
 
@@ -879,7 +879,7 @@ def test_connectivity_measure_check_vectorization_option(
     ["covariance", "correlation", "precision", "partial correlation"],
 )
 def test_connectivity_measure_check_inverse_transformation(
-    kind: str, signals: List[np.ndarray]
+    kind: str, signals: list[np.ndarray]
 ) -> None:
     """Test inverse transform.
 
@@ -915,7 +915,7 @@ def test_connectivity_measure_check_inverse_transformation(
     ["covariance", "correlation", "precision", "partial correlation"],
 )
 def test_connectivity_measure_check_inverse_transformation_discard_diag(
-    kind: str, signals: List[np.ndarray]
+    kind: str, signals: list[np.ndarray]
 ) -> None:
     """Test diagonal for inverse transform.
 
@@ -957,7 +957,7 @@ def test_connectivity_measure_check_inverse_transformation_discard_diag(
 
 
 def test_connectivity_measure_inverse_transform_tangent(
-    signals: List[np.ndarray],
+    signals: list[np.ndarray],
 ) -> None:
     """Test that for 'tangent' kind, covariance matrices are reconstructed.
 
@@ -1067,7 +1067,7 @@ def test_confounds_connectivity_measure_errors() -> None:
 
 
 def test_connectivity_measure_standardize(
-    signals: List[np.ndarray],
+    signals: list[np.ndarray],
 ) -> None:
     """Check warning is raised and then suppressed with setting standardize.
 

junifer/external/nilearn/tests/test_junifer_nifti_spheres_masker.py

@@ -4,7 +4,6 @@
 # License: AGPL
 
 import warnings
-from typing import List, Tuple
 
 import nibabel
 import numpy as np
@@ -361,10 +360,10 @@ def test_nifti_spheres_masker_io_shapes() -> None:
     ],
 )
 def test_junifer_and_nilearn_mean_agg_are_equal(
-    shape: Tuple[int, ...],
+    shape: tuple[int, ...],
     radius: float,
     allow_overlap: bool,
-    coords: List[Tuple[int, int, int]],
+    coords: list[tuple[int, int, int]],
 ) -> None:
     """Test junifer's masker behaves same as nilearn's when agg is mean.
 

junifer/markers/base.py

@@ -6,7 +6,7 @@
 
 from abc import ABC, abstractmethod
 from copy import deepcopy
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Optional, Union
 
 from ..pipeline import PipelineStepMixin, UpdateMetaMixin
 from ..typing import StorageLike
@@ -42,7 +42,7 @@ class BaseMarker(ABC, PipelineStepMixin, UpdateMetaMixin):
 
     def __init__(
         self,
-        on: Optional[Union[List[str], str]] = None,
+        on: Optional[Union[list[str], str]] = None,
         name: Optional[str] = None,
     ) -> None:
         # Check for missing mapping attribute
@@ -65,7 +65,7 @@ class BaseMarker(ABC, PipelineStepMixin, UpdateMetaMixin):
             raise_error(f"{self.name} cannot be computed on {wrong_on}")
         self._on = on
 
-    def validate_input(self, input: List[str]) -> List[str]:
+    def validate_input(self, input: list[str]) -> list[str]:
         """Validate input.
 
         Parameters
@@ -94,7 +94,7 @@ class BaseMarker(ABC, PipelineStepMixin, UpdateMetaMixin):
             )
         return [x for x in self._on if x in input]
 
-    def get_valid_inputs(self) -> List[str]:
+    def get_valid_inputs(self) -> list[str]:
         """Get valid data types for input.
 
         Returns
@@ -124,7 +124,7 @@ class BaseMarker(ABC, PipelineStepMixin, UpdateMetaMixin):
         return self._MARKER_INOUT_MAPPINGS[input_type][output_feature]
 
     @abstractmethod
-    def compute(self, input: Dict, extra_input: Optional[Dict] = None) -> Dict:
+    def compute(self, input: dict, extra_input: Optional[dict] = None) -> dict:
         """Compute.
 
         Parameters
@@ -155,7 +155,7 @@ class BaseMarker(ABC, PipelineStepMixin, UpdateMetaMixin):
         self,
         type_: str,
         feature: str,
-        out: Dict[str, Any],
+        out: dict[str, Any],
         storage: StorageLike,
     ) -> None:
         """Store.
@@ -178,9 +178,9 @@ class BaseMarker(ABC, PipelineStepMixin, UpdateMetaMixin):
 
     def _fit_transform(
         self,
-        input: Dict[str, Dict],
+        input: dict[str, dict],
         storage: Optional[StorageLike] = None,
-    ) -> Dict:
+    ) -> dict:
         """Fit and transform.
 
         Parameters

junifer/markers/brainprint.py

@@ -8,8 +8,6 @@ from pathlib import Path
 from typing import (
     Any,
     ClassVar,
-    Dict,
-    List,
     Optional,
     Union,
 )
@@ -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.
 
@@ -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/complexity_base.py

@@ -8,8 +8,6 @@ from typing import (
     TYPE_CHECKING,
     Any,
     ClassVar,
-    Dict,
-    List,
     Optional,
     Union,
 )
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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/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
@@ -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,7 +6,7 @@
 #          Synchon Mandal <s.mandal@fz-juelich.de>
 # License: AGPL
 
-from typing import Any, ClassVar, Dict, List, Optional, Union
+from typing import Any, ClassVar, Optional, Union
 
 import numpy as np
 
@@ -56,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
@@ -70,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/_afni_falff.py

@@ -9,7 +9,6 @@ from typing import (
     TYPE_CHECKING,
     ClassVar,
     Optional,
-    Tuple,
 )
 
 import nibabel as nib
@@ -55,7 +54,7 @@ class AFNIALFF(metaclass=Singleton):
         highpass: float,
         lowpass: float,
         tr: Optional[float],
-    ) -> Tuple["Nifti1Image", "Nifti1Image", Path, Path]:
+    ) -> tuple["Nifti1Image", "Nifti1Image", Path, Path]:
         """Compute ALFF + fALFF map.
 
         Parameters

junifer/markers/falff/_junifer_falff.py

@@ -9,7 +9,6 @@ from typing import (
     TYPE_CHECKING,
     ClassVar,
     Optional,
-    Tuple,
 )
 
 import nibabel as nib
@@ -52,7 +51,7 @@ class JuniferALFF(metaclass=Singleton):
         highpass: float,
         lowpass: float,
         tr: Optional[float],
-    ) -> Tuple["Nifti1Image", "Nifti1Image", Path, Path]:
+    ) -> tuple["Nifti1Image", "Nifti1Image", Path, Path]:
         """Compute ALFF + fALFF map.
 
         Parameters

junifer/markers/falff/falff_base.py

@@ -11,9 +11,7 @@ from typing import (
     TYPE_CHECKING,
     Any,
     ClassVar,
-    Dict,
     Optional,
-    Tuple,
 )
 
 from ...typing import ConditionalDependencies, MarkerInOutMappings
@@ -116,8 +114,8 @@ class ALFFBase(BaseMarker):
 
     def _compute(
         self,
-        input_data: Dict[str, Any],
-    ) -> Tuple["Nifti1Image", "Nifti1Image", Path, Path]:
+        input_data: dict[str, Any],
+    ) -> tuple["Nifti1Image", "Nifti1Image", Path, Path]:
         """Compute ALFF and fALFF.
 
         Parameters

junifer/markers/falff/falff_parcels.py

@@ -6,7 +6,7 @@
 #          Synchon Mandal <s.mandal@fz-juelich.de>
 # License: AGPL
 
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Optional, Union
 
 from ...api.decorators import register_marker
 from ...utils import logger
@@ -70,14 +70,14 @@ class ALFFParcels(ALFFBase):
 
     def __init__(
         self,
-        parcellation: Union[str, List[str]],
+        parcellation: Union[str, list[str]],
         using: str,
         highpass: float = 0.01,
         lowpass: float = 0.1,
         tr: Optional[float] = None,
         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:
         # Superclass init first to validate `using` parameter
@@ -95,9 +95,9 @@ class ALFFParcels(ALFFBase):
 
     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/falff/falff_spheres.py

@@ -6,7 +6,7 @@
 #          Synchon Mandal <s.mandal@fz-juelich.de>
 # License: AGPL
 
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Optional, Union
 
 from ...api.decorators import register_marker
 from ...utils import logger
@@ -85,8 +85,8 @@ class ALFFSpheres(ALFFBase):
         lowpass: float = 0.1,
         tr: Optional[float] = None,
         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:
         # Superclass init first to validate `using` parameter
@@ -106,9 +106,9 @@ class ALFFSpheres(ALFFBase):
 
     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