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.61',
      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',
          '*.sh',
          '*.txt',
          '*.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'
      )
