fix loss

src/otx/algo/detection/losses/accuracy.py

@@ -6,12 +6,16 @@
 
 from torch import nn
 
-
 if TYPE_CHECKING:
     from torch import Tensor
 
 
-def accuracy(pred: Tensor, target: Tensor, topk: int = 1, thresh: float | None = None):
+def accuracy(
+    pred: Tensor,
+    target: Tensor,
+    topk: int | tuple[int] = 1,
+    thresh: float | None = None,
+) -> list:
     """Calculate accuracy according to the prediction and target.
 
     Args:

src/otx/algo/detection/losses/focal_loss.py

@@ -14,7 +14,6 @@
 from .accuracy import accuracy
 from .utils import weight_reduce_loss
 
-
 if TYPE_CHECKING:
     from torch import Tensor
 
@@ -79,7 +78,8 @@ def py_focal_loss_with_prob(
     reduction: str = "mean",
     avg_factor: int | None = None,
 ):
-    """PyTorch version of `Focal Loss <https://arxiv.org/abs/1708.02002>`_.
+    """PyTorch version of `Focal Loss <https://arxiv.org/abs/1708.02002>`.
+
     Different from `py_sigmoid_focal_loss`, this function accepts probability
     as input.