Update test/test_transforms_v2_refactored.py

Co-authored-by: Nicolas Hug <[email protected]>

Author: pmeier

test/test_transforms_v2_refactored.py

@@ -1645,8 +1645,8 @@ def _transform(self, inpt, params):
 
     class PackedInputTransform(nn.Module):
         def forward(self, sample):
-            image, label = sample
-            return image, label
+            assert len(sample) == 2
+            return sample
 
     class UnpackedInputTransform(nn.Module):
         def forward(self, image, label):