forked from openvinotoolkit/datumaro
-
Notifications
You must be signed in to change notification settings - Fork 1
/
setup.py
114 lines (95 loc) · 3.5 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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
# Copyright (C) 2019-2022 Intel Corporation
#
# SPDX-License-Identifier: MIT
# ruff: noqa: E501
import os
import os.path as osp
import re
from distutils.util import strtobool
import setuptools
from pybind11.setup_helpers import Pybind11Extension, build_ext
from setuptools_rust import Binding, RustExtension
def find_version(project_dir=None):
if not project_dir:
project_dir = osp.dirname(osp.abspath(__file__))
file_path = osp.join(project_dir, "datumaro", "version.py")
with open(file_path, "r", encoding="utf-8") as version_file:
version_text = version_file.read()
# PEP440:
# https://www.python.org/dev/peps/pep-0440/#appendix-b-parsing-version-strings-with-regular-expressions
pep_regex = r"([1-9]\d*!)?(0|[1-9]\d*)(\.(0|[1-9]\d*))*((a|b|rc)(0|[1-9]\d*))?(\.post(0|[1-9]\d*))?(\.dev(0|[1-9]\d*))?"
version_regex = r"__version__\s*=\s*.(" + pep_regex + ")."
match = re.match(version_regex, version_text)
if not match:
raise RuntimeError("Failed to find version string in '%s'" % file_path)
version = version_text[match.start(1) : match.end(1)]
return version
CORE_REQUIREMENTS_FILE = "requirements-core.txt"
DEFAULT_REQUIREMENTS_FILE = "requirements-default.txt"
def parse_requirements(filename=CORE_REQUIREMENTS_FILE):
with open(filename, "r", encoding="utf-8") as fh:
return fh.readlines()
CORE_REQUIREMENTS = parse_requirements(CORE_REQUIREMENTS_FILE)
if strtobool(os.getenv("DATUMARO_HEADLESS", "0").lower()):
CORE_REQUIREMENTS.append("opencv-python-headless")
else:
CORE_REQUIREMENTS.append("opencv-python")
DEFAULT_REQUIREMENTS = parse_requirements(DEFAULT_REQUIREMENTS_FILE)
with open("README.md", "r", encoding="utf-8") as fh:
long_description = fh.read()
ext_modules = [
Pybind11Extension(
"datumaro._capi",
["src/datumaro/capi/pybind.cpp"],
define_macros=[("VERSION_INFO", find_version("./src"))],
extra_compile_args=["-O3"],
),
]
setuptools.setup(
name="datumaro",
version=find_version("./src"),
author="Intel",
author_email="[email protected]",
description="Dataset Management Framework (Datumaro)",
long_description=long_description,
long_description_content_type="text/markdown",
url="https://github.com/openvinotoolkit/datumaro",
package_dir={"": "src"},
packages=setuptools.find_packages(where="src", include=["datumaro*"]),
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
],
python_requires=">=3.9",
install_requires=CORE_REQUIREMENTS,
extras_require={
"tf": ["tensorflow"],
"tfds": ["tensorflow-datasets<4.9.3"],
"tf-gpu": ["tensorflow-gpu"],
"torch": ["torch", "torchvision"],
"default": DEFAULT_REQUIREMENTS,
},
ext_modules=ext_modules,
entry_points={
"console_scripts": [
"datum=datumaro.cli.__main__:main",
],
},
cmdclass={"build_ext": build_ext},
package_data={
"datumaro.plugins.synthetic_data": ["background_colors.txt"],
"datumaro.plugins.openvino_plugin.samples": ["coco.class", "imagenet.class"],
},
include_package_data=True,
rust_extensions=[
RustExtension(
"datumaro.rust_api",
path=osp.join("rust", "Cargo.toml"),
debug=False,
binding=Binding.PyO3,
),
],
# rust extensions are not zip safe, just like C-extensions.
zip_safe=False,
)