Skip to content

Commit ca115ad

Browse files
committed
Fix lint
1 parent eefa9f8 commit ca115ad

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

test/test_transforms_v2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5592,7 +5592,7 @@ def test_clamping_mode(self, rotated, constructor_clamping_mode, clamping_mode,
55925592
out = fn(boxes, clamping_mode=clamping_mode)
55935593

55945594
clamping_mode_prevailing = constructor_clamping_mode if clamping_mode is None else clamping_mode
5595-
if clamping_mode_prevailing == None:
5595+
if clamping_mode_prevailing is None:
55965596
assert_equal(boxes, out) # should be a pass-through
55975597
else:
55985598
assert_equal(out, expected_clamped_output)
@@ -5643,7 +5643,7 @@ def transform(self, inpt, _):
56435643

56445644
# assert that the output boxes clamping_mode is the one set by the last SetClampingMode.
56455645
# ClampBoundingBoxes doesn't set clamping_mode.
5646-
assert out_boxes.clamping_mode == None
5646+
assert out_boxes.clamping_mode is None
56475647

56485648

56495649
class TestClampKeyPoints:

torchvision/transforms/v2/_meta.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from typing import Any, Optional, Union
1+
from typing import Any, Union
22

33
from torchvision import tv_tensors
44
from torchvision.transforms.v2 import functional as F, Transform

0 commit comments

Comments
 (0)