from setuptools import setup, find_packages

setup(
    name="easy_smr",
    description='Easy Sagemaker Ops for R projects',
    long_description="""This package makes it easier to work with Sagemaker by enabling rapid prototyping with local training, processing and deployment.
And correspondingly training, processing and deployment on cloud.
This is very much an experimental package and API is likely to evolve and may break.
Recommended to validate before updating.
    """,
    author='Prateek',
    author_email='prteek@icloud.com',
    version="0.1.11",
    python_requires='>=3.11',
    packages=find_packages(where='.'),
    package_data={
        'easy_smr': [
            'template/easy_smr_base/*.sh',
            'template/easy_smr_base/Dockerfile',
            'template/easy_smr_base/.dockerignore',
            'template/easy_smr_base/training/train',
            'template/easy_smr_base/training/*.R',
            'template/easy_smr_base/processing/.gitkeep',
            'template/easy_smr_base/prediction/*.R',
            'template/easy_smr_base/prediction/serve',
            'template/easy_smr_base/local_test/*.sh',
            'template/easy_smr_base/local_test/test_dir/output/.gitkeep',
            'template/easy_smr_base/local_test/test_dir/model/.gitkeep',
            'template/easy_smr_base/local_test/test_dir/input/data/training/.gitkeep',
            'template/easy_smr_base/renv/.gitignore',
            'template/easy_smr_base/renv/activate.R',
            'template/easy_smr_base/renv/settings.json',
            'template/easy_smr_base/renv.lock',
            'template/easy_smr_base/.Rprofile'
        ]
    },
    install_requires=[
        'click>=8.1.7, <8.1.99',
        'docker>=7.1.0, <7.2.0',
        'sagemaker>=2.226.0, <2.228.0',
    ],
    entry_points={
        'console_scripts': [
            'easy_smr=easy_smr.__main__:cli',
        ],
    }
)