forked from activeloopai/deeplake
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
116 lines (95 loc) · 3.32 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
115
116
import os
import sys
import re
import platform
from setuptools import find_packages, setup
project_name = "deeplake"
this_directory = os.path.abspath(os.path.dirname(__file__))
with open(os.path.join(this_directory, "deeplake/requirements/common.txt")) as f:
requirements = f.readlines()
with open(os.path.join(this_directory, "deeplake/requirements/tests.txt")) as f:
tests = f.readlines()
with open(os.path.join(this_directory, "README.md"), encoding="utf-8") as f:
long_description = f.read()
req_map = {
b: a
for a, b in (
re.findall(r"^(([^!=<>~]+)(?:[!=<>~].*)?$)", x.strip("\n"))[0]
for x in requirements
)
}
# Add optional dependencies to this dict without version. Version should be specified in requirements.txt
extras = {
"audio": ["av"],
"video": ["av"],
"av": ["av"],
"gcp": ["google-cloud-storage", "google-auth", "google-auth-oauthlib"],
"azure": ["azure-cli", "azure-identity", "azure-storage-blob"],
"dicom": ["pydicom", "nibabel"],
"medical": ["pydicom", "nibabel"],
"visualizer": ["IPython", "flask"],
"gdrive": [
"google-api-python-client",
"oauth2client",
"google-auth",
"google-auth-oauthlib",
],
"point_cloud": ["laspy"],
}
def libdeeplake_available():
py_ver = sys.version_info
if sys.platform == "linux":
if py_ver >= (3, 6) and py_ver <= (3, 12):
return True
if sys.platform == "darwin":
mac_ver = list(map(int, platform.mac_ver()[0].split(".")))
if (
(mac_ver[0] > 10 or mac_ver[0] == 10 and mac_ver[1] >= 12)
and py_ver >= (3, 7)
and py_ver < (3, 12)
):
return True
return False
all_extras = {r for v in extras.values() for r in v}
install_requires = [req_map[r] for r in req_map if r not in all_extras]
extras_require = {k: [req_map[r] for r in v] for k, v in extras.items()}
extras_require["all"] = [req_map[r] for r in all_extras]
if libdeeplake_available():
libdeeplake = "libdeeplake==0.0.95"
extras_require["enterprise"] = [libdeeplake, "pyjwt"]
extras_require["all"].append(libdeeplake)
install_requires.append(libdeeplake)
init_file = os.path.join(project_name, "__init__.py")
def get_property(prop):
result = re.search(
# find variable with name `prop` in the __init__.py file
rf'{prop}\s*=\s*[\'"]([^\'"]*)[\'"]',
open(init_file).read(),
)
return result.group(1)
config = {
"name": project_name,
"version": get_property("__version__"),
"description": "Activeloop Deep Lake",
"long_description": long_description,
"long_description_content_type": "text/markdown",
"author": "activeloop.ai",
"author_email": "support@activeloop.ai",
"packages": find_packages(),
"install_requires": install_requires,
"extras_require": extras_require,
"tests_require": tests,
"include_package_data": True,
"zip_safe": False,
"entry_points": {"console_scripts": ["activeloop = deeplake.cli.commands:cli"]},
"dependency_links": [],
"project_urls": {
"Documentation": "https://docs.activeloop.ai/",
"Source": "https://github.com/activeloopai/deeplake",
},
"license": "MPL-2.0",
"classifiers": [
"License :: OSI Approved :: Mozilla Public License 2.0 (MPL 2.0)",
],
}
setup(**config)