Removed torch module as already v0.3.0 and fixed flake8

albumentations-team · Sep 2, 2019 · 1983417 · 1983417
1 parent 7d20e1d
commit 1983417
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/albumentations/pytorch/transforms.py b/albumentations/pytorch/transforms.py
@@ -10,8 +10,8 @@
 
 class ToTensor(BasicTransform):
     """Convert image and mask to `torch.Tensor`.
-
     """
+
     def __init__(self):
         super(ToTensor, self).__init__(always_apply=True)
 

diff --git a/albumentations/torch/__init__.py b/albumentations/torch/__init__.py