From 87a9437057df8a7550aa3b3eaf840871cc0d5cef Mon Sep 17 00:00:00 2001
From: Clement Walter <clement.walter@ibm.com>
Date: Wed, 14 Apr 2021 09:21:15 +0200
Subject: [PATCH 1/2] Fix requirements and remove deprecated unittest2

---
 requirements.txt             | 2 +-
 tests/test_axis_mirroring.py | 4 +---
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 4eecc7c..073dd21 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,7 +1,7 @@
 pillow>=7.1.2
 threadpoolctl
-scikit-learn
 numpy>=1.10.2
 scipy
 scikit-image
 scikit-learn
+future
diff --git a/tests/test_axis_mirroring.py b/tests/test_axis_mirroring.py
index 1b4598c..19465e2 100644
--- a/tests/test_axis_mirroring.py
+++ b/tests/test_axis_mirroring.py
@@ -12,8 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import unittest
-
-import unittest2
 import numpy as np
 from skimage import data
 
@@ -22,7 +20,7 @@
 from batchgenerators.transforms.spatial_transforms import MirrorTransform
 
 
-class TestMirrorAxis(unittest2.TestCase):
+class TestMirrorAxis(unittest.TestCase):
     def setUp(self):
         self.seed = 1234
 

From bec17e22e536404ebaf504b60a0e85529aca3556 Mon Sep 17 00:00:00 2001
From: Clement Walter <clement.walter@ibm.com>
Date: Wed, 14 Apr 2021 09:26:07 +0200
Subject: [PATCH 2/2] Remove unittest2 from setup.py

---
 setup.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/setup.py b/setup.py
index ef0def0..369d7db 100755
--- a/setup.py
+++ b/setup.py
@@ -15,7 +15,6 @@
             "scikit-image",
             "scikit-learn",
             "future",
-            "unittest2",
             "threadpoolctl"
       ],
       keywords=['deep learning', 'image segmentation', 'image classification', 'medical image analysis',