fix typing

Warvito · Mar 29, 2023 · 1b8ed97 · 1b8ed97
1 parent e3545ae
commit 1b8ed97
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/monai/losses/ssim_loss.py b/monai/losses/ssim_loss.py
@@ -37,8 +37,8 @@ def __init__(
         spatial_dims: int,
         data_range: float = 1.0,
         kernel_type: KernelType | str = KernelType.GAUSSIAN,
-        kernel_size: int | Sequence[int, ...] = 11,
-        kernel_sigma: float | Sequence[float, ...] = 1.5,
+        kernel_size: int | Sequence[int] = 11,
+        kernel_sigma: float | Sequence[float] = 1.5,
         k1: float = 0.01,
         k2: float = 0.03,
         reduction: LossReduction | str = LossReduction.MEAN,

diff --git a/monai/metrics/regression.py b/monai/metrics/regression.py
@@ -271,8 +271,8 @@ def __init__(
         spatial_dims: int,
         data_range: float = 1.0,
         kernel_type: KernelType | str = KernelType.GAUSSIAN,
-        kernel_size: int | Sequence[int, ...] = 11,
-        kernel_sigma: float | Sequence[float, ...] = 1.5,
+        kernel_size: int | Sequence[int] = 11,
+        kernel_sigma: float | Sequence[float] = 1.5,
         k1: float = 0.01,
         k2: float = 0.03,
         reduction: MetricReduction | str = MetricReduction.MEAN,
@@ -339,7 +339,7 @@ def _compute_metric(self, y_pred: torch.Tensor, y: torch.Tensor) -> torch.Tensor
 
 
 def _gaussian_kernel(
-    spatial_dims: int, num_channels: int, kernel_size: Sequence[int, ...], kernel_sigma: Sequence[float, ...]
+    spatial_dims: int, num_channels: int, kernel_size: Sequence[int], kernel_sigma: Sequence[float]
 ) -> torch.Tensor:
     """Computes 2D or 3D gaussian kernel.
 
@@ -384,8 +384,8 @@ def compute_ssim_and_cs(
     spatial_dims: int,
     data_range: float = 1.0,
     kernel_type: KernelType | str = KernelType.GAUSSIAN,
-    kernel_size: Sequence[int, ...] = 11,
-    kernel_sigma: Sequence[float, ...] = 1.5,
+    kernel_size: Sequence[int] = 11,
+    kernel_sigma: Sequence[float] = 1.5,
     k1: float = 0.01,
     k2: float = 0.03,
 ) -> tuple[torch.Tensor, torch.Tensor]: