Fix pre-commit.ci

zhiqwang · Oct 24, 2021 · b9dbbb2 · b9dbbb2
1 parent ce91bce
commit b9dbbb2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolort/models/box_head.py b/yolort/models/box_head.py
@@ -226,7 +226,7 @@ def build_targets(
         targets: Tensor,
         head_outputs: List[Tensor],
         anchor_grids: Tensor,
-    ) -> Tuple[List[Tensor], List[Tensor], List[Tuple[Tensor, Tensor, Tensor, Tensor]], List[Tensor],]:
+    ) -> Tuple[List[Tensor], List[Tensor], List[Tuple[Tensor, Tensor, Tensor, Tensor]], List[Tensor]]:
         device = targets.device
         num_anchors = self.num_anchors