forked from bstriner/aws-sagemaker-remote
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
42 lines (42 loc) · 1.33 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
from setuptools import find_packages, setup
import os
with open(os.path.abspath(os.path.join(__file__, '../README.rst')), encoding='utf-8') as f:
long_description = f.read()
setup(name='aws-sagemaker-remote',
version='0.0.71',
author='Ben Striner',
author_email="bstriner@gmail.com",
url='https://github.com/bstriner/aws-sagemaker-remote',
install_requires=[
'sagemaker',
'sagemaker-experiments',
'click'
],
# package_data={'aws_sagemaker_remote': [
# 'ecr/inference/latest/Dockerfile',
# 'ecr/processing/latest/Dockerfile',
# 'ecr/training/latest/Dockerfile',
# 'ecr/training/gpu/Dockerfile',
# 'Dockerfile',
# '*Dockerfile',
# '*.sh',
# '*.txt',
# '*.types',
# '*.yaml',
# '*.properties',
# '*.yml',
# '*.bat',
# '*.js',
# '*.json'
# ]},
include_package_data=True,
entry_points={
'console_scripts': [
'aws-sagemaker-remote=aws_sagemaker_remote.cli:cli'
]
},
description="Simplify running processing and training remotely on AWS SageMaker",
packages=find_packages(),
long_description=long_description,
long_description_content_type='text/x-rst'
)