[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
kornia · Nov 29, 2022 · 7c36635 · 7c36635
1 parent 4cad3b0
commit 7c36635
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/kornia/contrib/face_detection.py b/kornia/contrib/face_detection.py
@@ -217,7 +217,6 @@ def forward(self, image: torch.Tensor) -> List[torch.Tensor]:
 
         Return:
             List[torch.Tensor]: list with the boxes found on each image. :math:`Bx(N,15)`.
-
         """
         img = self.preprocess(image)
         out = self.model(img)

diff --git a/test/test_contrib.py b/test/test_contrib.py
@@ -583,7 +583,7 @@ def test_valid(self, batch_size, device, dtype):
         face_detection = kornia.contrib.FaceDetector().to(device, dtype)
         dets = face_detection(img)
         assert isinstance(dets, list)
-        assert len(dets) == batch_size # same as the number of images
+        assert len(dets) == batch_size  # same as the number of images
         assert isinstance(dets[0], torch.Tensor)
         assert dets[0].shape[0] >= 0  # number of detections
         assert dets[0].shape[1] == 15  # dims of each detection