FilesExpand file tree

 master

/

setup.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

113 lines (103 loc) · 3.2 KB

 master

/

setup.py

Copy path

Top

File metadata and controls

• Code
• Blame

113 lines (103 loc) · 3.2 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

# Copyright 2017-2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.

#

# Licensed under the Apache License, Version 2.0 (the "License"). You

# may not use this file except in compliance with the License. A copy of

# the License is located at

#

# http://aws.amazon.com/apache2.0/

#

# or in the "license" file accompanying this file. This file is

# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF

# ANY KIND, either express or implied. See the License for the specific

# language governing permissions and limitations under the License.

"""Placeholder docstring"""

from __future__ import absolute_import

import os

from glob import glob

from setuptools import setup, find_packages

def read(fname):

"""

Args:

fname:

"""

return open(os.path.join(os.path.dirname(__file__), fname)).read()

def read_version():

return read("VERSION").strip()

# Declare minimal set for installation

required_packages = [

"attrs",

"boto3>=1.16.32",

"google-pasta",

"numpy>=1.9.0",

"protobuf>=3.1",

"protobuf3-to-dict>=0.1.5",

"smdebug_rulesconfig==1.0.1",

"importlib-metadata>=1.4.0",

"packaging>=20.0",

"pandas",

"pathos",

]

# Specific use case dependencies

extras = {

"local": [

"urllib3>=1.21.1,!=1.25,!=1.25.1",

"docker-compose>=1.25.2",

"PyYAML>=5.3, <6", # PyYAML version has to match docker-compose requirements

],

"scipy": ["scipy>=0.19.0"],

}

# Meta dependency groups

extras["all"] = [item for group in extras.values() for item in group]

# Tests specific dependencies (do not need to be included in 'all')

extras["test"] = (

[

extras["all"],

"tox",

"flake8",

"pytest<6.1.0",

"pytest-cov",

"pytest-rerunfailures",

"pytest-timeout",

"pytest-xdist",

"mock",

"contextlib2",

"awslogs",

"black",

"stopit==1.1.2",

"apache-airflow==1.10.11",

"fabric>=2.0",

"requests>=2.20.0, <3",

"sagemaker-experiments",

],

)

setup(

name="sagemaker",

version=read_version(),

description="Open source library for training and deploying models on Amazon SageMaker.",

packages=find_packages("src"),

package_dir={"": "src"},

py_modules=[os.path.splitext(os.path.basename(path))[0] for path in glob("src/*.py")],

include_package_data=True,

long_description=read("README.rst"),

author="Amazon Web Services",

url="https://github.com/aws/sagemaker-python-sdk/",

license="Apache License 2.0",

keywords="ML Amazon AWS AI Tensorflow MXNet",

classifiers=[

"Development Status :: 5 - Production/Stable",

"Intended Audience :: Developers",

"Natural Language :: English",

"License :: OSI Approved :: Apache Software License",

"Programming Language :: Python",

"Programming Language :: Python :: 3.6",

"Programming Language :: Python :: 3.7",

"Programming Language :: Python :: 3.8",

],

install_requires=required_packages,

extras_require=extras,

entry_points={

"console_scripts": [

"sagemaker-upgrade-v2=sagemaker.cli.compatibility.v2.sagemaker_upgrade_v2:main",

]

},

)