import Sequence from typing

Signed-off-by: ytl0623 <[email protected]>

Author: ytl0623

monai/losses/focal_loss.py

@@ -12,7 +12,7 @@
 from __future__ import annotations
 
 import warnings
-from typing import Optional, Sequence
+from typing import Optional, Sequence as SequenceType
 
 import torch
 import torch.nn.functional as F
@@ -207,7 +207,7 @@ def forward(self, input: torch.Tensor, target: torch.Tensor) -> torch.Tensor:
 
 
 def softmax_focal_loss(
-    input: torch.Tensor, target: torch.Tensor, gamma: float = 2.0, alpha: float | Sequence[float] | None = None
+    input: torch.Tensor, target: torch.Tensor, gamma: float = 2.0, alpha: float | SequenceType[float] | None = None
 ) -> torch.Tensor:
     """
     FL(pt) = -alpha * (1 - pt)**gamma * log(pt)
@@ -240,7 +240,7 @@ def softmax_focal_loss(
 
 
 def sigmoid_focal_loss(
-    input: torch.Tensor, target: torch.Tensor, gamma: float = 2.0, alpha: float | Sequence[float] | None = None
+    input: torch.Tensor, target: torch.Tensor, gamma: float = 2.0, alpha: float | SequenceType[float] | None = None
 ) -> torch.Tensor:
     """
     FL(pt) = -alpha * (1 - pt)**gamma * log(pt)