[fbsync] [proto] Compose transform keeps BC (#6391)

Summary:
* [proto] Compose keeps BC

* Compose -> Compose(Transform)

Reviewed By: datumbox

Differential Revision: D38824249

fbshipit-source-id: 294949efaee4cb01be5a958799deac9f3018d9e4

Author: NicolasHug

test/test_prototype_transforms.py

@@ -1083,3 +1083,22 @@ def test__transform(self, inpt_type, mocker):
             fn.call_count == 0
         else:
             fn.assert_called_once_with(inpt)
+
+
+class TestCompose:
+    def test_assertions(self):
+        with pytest.raises(TypeError, match="Argument transforms should be a sequence of callables"):
+            transforms.Compose(123)
+
+    @pytest.mark.parametrize(
+        "trfms",
+        [
+            [transforms.Pad(2), transforms.RandomCrop(28)],
+            [lambda x: 2.0 * x],
+        ],
+    )
+    def test_ctor(self, trfms):
+        c = transforms.Compose(trfms)
+        inpt = torch.rand(1, 3, 32, 32)
+        output = c(inpt)
+        assert isinstance(output, torch.Tensor)

torchvision/prototype/transforms/_container.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, List, Optional
+from typing import Any, Callable, Dict, List, Optional, Sequence
 
 import torch
 from torchvision.prototype.transforms import Transform
@@ -7,11 +7,11 @@
 
 
 class Compose(Transform):
-    def __init__(self, *transforms: Transform) -> None:
+    def __init__(self, transforms: Sequence[Callable]) -> None:
         super().__init__()
+        if not isinstance(transforms, Sequence):
+            raise TypeError("Argument transforms should be a sequence of callables")
         self.transforms = transforms
-        for idx, transform in enumerate(transforms):
-            self.add_module(str(idx), transform)
 
     def forward(self, *inputs: Any) -> Any:
         sample = inputs if len(inputs) > 1 else inputs[0]