Skip to content

Commit 2e2ea83

Browse files
fix linting
1 parent b22688a commit 2e2ea83

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

torchvision/transforms/v2/_meta.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -85,9 +85,7 @@ def __init__(self, clamping_mode: CLAMPING_MODE_TYPE) -> None:
8585

8686
_transformed_types = (tv_tensors.BoundingBoxes, tv_tensors.KeyPoints)
8787

88-
def transform(
89-
self, inpt: tv_tensors.BoundingBoxes | tv_tensors.KeyPoints, params: dict[str, Any]
90-
) -> tv_tensors.BoundingBoxes | tv_tensors.KeyPoints:
88+
def transform(self, inpt: tv_tensors.TVTensor, params: dict[str, Any]) -> tv_tensors.TVTensor:
9189
out: tv_tensors.TVTensor = inpt.clone() # type: ignore[assignment]
92-
out.clamping_mode = self.clamping_mode
90+
out.clamping_mode = self.clamping_mode # type: ignore[assignment]
9391
return out

0 commit comments

Comments
 (0)