(Update) Updated code to run with albumentations version 1.0.3. (#927)

* (Update) Updated code to run with albumentations version 1.0.3. * Fixes using = instead of the correct == in the setup.cfg * Updated version requirement for albumentations to >=1.0.3,<1.1 Co-authored-by: frederik <fstroth>
airctic · Sep 23, 2021 · 3a426a3 · 3a426a3
1 parent 9f23f35
commit 3a426a3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -30,7 +30,7 @@ install_requires =
     fastcore >=1.3.0,<1.4
     tqdm >=4.49.0,<5
     opencv-python >=4.1.1,<5
-    albumentations >=0.4.6,<0.6
+    albumentations >=1.0.3,<1.1
     matplotlib >=3.2.2,<4
     pycocotools>=2.0.2,<3
     requests >=2.23.0,<3

diff --git a/tests/transforms/test_albu_transform.py b/tests/transforms/test_albu_transform.py
@@ -3,7 +3,7 @@
 )
 import pytest
 from icevision.all import *
-from albumentations.augmentations.transforms import (
+from albumentations import (
     LongestMaxSize,
     Normalize,
     PadIfNeeded,