-
-
Notifications
You must be signed in to change notification settings - Fork 1.6k
/
setup.py
45 lines (38 loc) · 1.4 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
import re
from pkg_resources import DistributionNotFound, get_distribution
from setuptools import setup, find_packages
INSTALL_REQUIRES = [
"numpy>=1.24.4",
"scipy>=1.10.0",
"PyYAML",
"typing-extensions>=4.9.0; python_version<'3.10'",
"pydantic>=2.9.2",
"albucore==0.0.21",
"eval-type-backport",
]
MIN_OPENCV_VERSION = "4.9.0.80"
CHOOSE_INSTALL_REQUIRES = [
(
(f"opencv-python>={MIN_OPENCV_VERSION}", f"opencv-contrib-python>={MIN_OPENCV_VERSION}", f"opencv-contrib-python-headless>={MIN_OPENCV_VERSION}"),
f"opencv-python-headless>={MIN_OPENCV_VERSION}",
),
]
def choose_requirement(mains: tuple[str, ...], secondary: str) -> str:
chosen = secondary
for main in mains:
try:
name = re.split(r"[!<>=]", main)[0]
get_distribution(name)
chosen = main
break
except DistributionNotFound:
pass
return chosen
def get_install_requirements(install_requires: list[str], choose_install_requires: list[tuple[tuple[str, ...], str]]) -> list[str]:
for mains, secondary in choose_install_requires:
install_requires.append(choose_requirement(mains, secondary))
return install_requires
setup(
packages=find_packages(exclude=["tests", "tools", "benchmark"], include=['albumentations*']),
install_requires=get_install_requirements(INSTALL_REQUIRES, CHOOSE_INSTALL_REQUIRES),
)