Fixed broken test_random_choice

test/test_transforms_v2.py

@@ -1359,7 +1359,7 @@ def test_ctor(self, transform_cls, trfms):
 
 class TestRandomChoice:
     def test_assertions(self):
-        with pytest.raises(ValueError, match="The number of probabilities doesn't match the number of transforms"):
+        with pytest.raises(ValueError, match="Length of p doesn't match the number of transforms"):
             transforms.RandomChoice([transforms.Pad(2), transforms.RandomCrop(28)], p=[1])
 
 

test/test_transforms_v2_consistency.py

@@ -822,7 +822,7 @@ def test_random_choice(self, probabilities):
                 v2_transforms.Resize(256),
                 legacy_transforms.CenterCrop(224),
             ],
-            probabilities=probabilities,
+            p=probabilities,
         )
         legacy_transform = legacy_transforms.RandomChoice(
             [

torchvision/transforms/v2/_container.py

@@ -139,7 +139,7 @@ def __init__(
             p = [1] * len(transforms)
         elif len(p) != len(transforms):
             raise ValueError(
-                f"The number of p doesn't match the number of transforms: " f"{len(p)} != {len(transforms)}"
+                f"Length of p doesn't match the number of transforms: " f"{len(p)} != {len(transforms)}"
             )
 
         super().__init__()