split extra dependencies

Digital-Dermatology · Oct 12, 2024 · 6416d21 · 6416d21
1 parent ff735e8
commit 6416d21
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -16,7 +16,9 @@ WORKDIR /workspace
 RUN mkdir /assets
 
 COPY requirements.txt /assets/requirements.txt
+COPY requirements.extras.txt /assets/requirements.extras.txt
 RUN pip install -r /assets/requirements.txt --upgrade --no-cache-dir
+RUN pip install -r /assets/requirements.extras.txt --upgrade --no-cache-dir
 
 COPY . /workspace/
 RUN git config --global --add safe.directory '*'

diff --git a/requirements.extras.txt b/requirements.extras.txt
@@ -0,0 +1 @@
+annoy
diff --git a/requirements.txt b/requirements.txt
@@ -22,4 +22,3 @@ codecov
 jupyter
 loguru
 memory-profiler
-annoy
diff --git a/setup.py b/setup.py
@@ -31,7 +31,7 @@ def parse_requirements(filename):
     name=PACKAGE_NAME,
     packages=proj_packages,
     package_dir={PACKAGE_NAME: SOURCE_DIRECTORY},
-    version="0.0.29",
+    version="0.0.30",
     author="Fabian Groeger",
     author_email="fabian.groeger@unibas.ch",
     description="A holistic self-supervised data cleaning strategy to detect irrelevant samples, near duplicates and label errors.",
@@ -41,6 +41,7 @@ def parse_requirements(filename):
     python_requires=">=3.6",
     install_requires=parse_requirements("requirements.txt"),
     setup_requires=parse_requirements("requirements.txt"),
+    extras_require={"approximate_nn": ["annoy"]},
     classifiers=[
         "Programming Language :: Python :: 3",
         "Operating System :: OS Independent",
-Original file line number
+Diff line change
@@ Expand Up / @@ -22,4 +22,3 @@ codecov @@
     jupyter
     loguru
     memory-profiler
-    annoy