Update test_ops.py

Author: kkaiwwana

test/test_ops.py

@@ -1488,17 +1488,17 @@ def test_distance_iou_loss(self, dtype, device):
     def test_empty_distance_iou_inputs(self, dtype, device):
         assert_empty_loss(ops.distance_box_iou_loss, dtype, device)
 
-        
+
 class TestLastLevelMaxPool2d:
     def _generate_fpn_feature_maps(self, **kwargs) -> Tuple[List[Tensor], List[str]]:
         feature_maps = [
             torch.rand(16, 3, 240, 320),
             torch.rand(16, 3, 120, 160),
             torch.rand(16, 3, 60, 80),
-            torch.rand(16, 3, 30, 40)
+            torch.rand(16, 3, 30, 40),
         ]
-        names = ['0', '1', '2', '3']
-        
+        names = ["0", "1", "2", "3"]
+
         return feature_maps, names
 
     def test_lastlevel_maxpool2d(self) -> None:
@@ -1514,7 +1514,7 @@ def test_lastlevel_maxpool2d(self) -> None:
         tol = 1e-3 if feature_maps[-1].dtype is torch.half else 1e-5
         torch.testing.assert_close(expected_pooled_featuremap, results[-1], atol=tol, rtol=tol)
 
-        
+
 class TestFocalLoss:
     def _generate_diverse_input_target_pair(self, shape=(5, 2), **kwargs):
         def logit(p):