FilesExpand file tree

main

/

setup.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

116 lines (95 loc) · 3.32 KB

main

/

setup.py

Copy path

Top

File metadata and controls

• Code
• Blame

116 lines (95 loc) · 3.32 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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.83"

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)