Skip to content

Commit

Permalink
enforce at least one transform
Browse files Browse the repository at this point in the history
  • Loading branch information
pmeier committed Jul 25, 2023
1 parent 4bcb488 commit b99647b
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions torchvision/transforms/v2/_container.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,14 +43,12 @@ def __init__(self, transforms: Sequence[Callable]) -> None:
super().__init__()
if not isinstance(transforms, Sequence):
raise TypeError("Argument transforms should be a sequence of callables")
elif not transforms:
raise ValueError("Pass at least one transform")
self.transforms = transforms

def forward(self, *inputs: Any) -> Any:
needs_unpacking = len(inputs) > 1

if not self.transforms:
return inputs if needs_unpacking else inputs[0]

for transform in self.transforms:
outputs = transform(*inputs)
inputs = outputs if needs_unpacking else (outputs,)
Expand Down

0 comments on commit b99647b

Please sign in to comment.