PyTorch fix.

Author: mdbloice

Augmentor/Pipeline.py

@@ -527,6 +527,7 @@ def _transform(image):
             for operation in self.operations:
                 r = round(random.uniform(0, 1), 1)
                 if r <= operation.probability:
+                    image = [image]
                     image = operation.perform_operation(image)
 
             return image

tests/test_torch_transform.py

@@ -23,5 +23,4 @@ def test_torch_transform():
     ])
 
     assert red != transforms(red)
-    red = [red]
     assert g.perform_operation(red) == transforms(red)