minor bug fix

requirements.txt

@@ -23,7 +23,6 @@ flake8==3.7.9
 future==0.18.2
 identify==1.4.7
 idna==2.8
-imageio==2.6.1
 imgaug==0.2.6
 importlib-metadata==0.23
 Jinja2==2.10.3

setup.py

@@ -0,0 +1,22 @@
+import setuptools
+
+with open("README.md", "r") as fh:
+    long_description = fh.read()
+
+setuptools.setup(
+    name="albumentations-demo", # Replace with your own username
+    version="0.0.1",
+    author="Ilya Larchenko",
+    author_email="[email protected]",
+    description="Service for demonstration of ",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    url="https://github.com/pypa/sampleproject",
+    packages=setuptools.find_packages(),
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+    ],
+    python_requires='>=3.6',
+)

tests/test_utils.py

@@ -0,0 +1,10 @@
+# test
+
+from src.utils import get_images_list
+
+
+def test_get_images_list():
+    images_list = get_images_list("images")
+    assert isinstance(images_list, list)
+    assert len(images_list) > 0
+    assert isinstance(images_list[0], str)