From 3a426a3a96438891a54c2ececa0d252941f8c6a1 Mon Sep 17 00:00:00 2001
From: fstroth <frstrothmann@gmail.com>
Date: Thu, 23 Sep 2021 14:29:14 +0200
Subject: [PATCH] (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>
---
 setup.cfg                               | 2 +-
 tests/transforms/test_albu_transform.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.cfg b/setup.cfg
index 982ea666d..73e6ffdc6 100644
--- 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
index bfb2f557a..0c85df040 100644
--- 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,