Skip to content

Commit 6c481cf

Browse files
committed
Lint
1 parent 0e1caea commit 6c481cf

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

test/test_transforms_v2.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5444,6 +5444,7 @@ def test_errors(self):
54445444
def test_transform(self):
54455445
check_transform(transforms.ClampBoundingBoxes(), make_bounding_boxes())
54465446

5447+
54475448
class TestClampKeyPoints:
54485449
@pytest.mark.parametrize("dtype", [torch.int64, torch.float32])
54495450
@pytest.mark.parametrize("device", cpu_and_cuda())
@@ -5472,7 +5473,6 @@ def test_transform(self):
54725473
check_transform(transforms.ClampKeyPoints(), make_keypoints())
54735474

54745475

5475-
54765476
class TestInvert:
54775477
@pytest.mark.parametrize("dtype", [torch.uint8, torch.int16, torch.float32])
54785478
@pytest.mark.parametrize("device", cpu_and_cuda())
@@ -7196,6 +7196,7 @@ def test_no_valid_input(self, query):
71967196
with pytest.raises(TypeError, match="No image"):
71977197
query(["blah"])
71987198

7199+
71997200
@pytest.mark.parametrize(
72007201
"boxes",
72017202
[

torchvision/transforms/v2/_meta.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ class ClampBoundingBoxes(Transform):
3535
def transform(self, inpt: tv_tensors.BoundingBoxes, params: dict[str, Any]) -> tv_tensors.BoundingBoxes:
3636
return F.clamp_bounding_boxes(inpt) # type: ignore[return-value]
3737

38+
3839
class ClampKeyPoints(Transform):
3940
"""Clamp keypoints to their corresponding image dimensions.
4041

0 commit comments

Comments
 (0)