from setuptools import setup, find_packages

setup(
    name='my_project',
    description='world piece, etc',
    version='0.0.1',
    packages=find_packages(exclude=['test', '*.test', '*.test.*']),
    include_package_data=True,
    install_requires=[
        'numpy==1.17.2',
    ],
    extras_require={
        'cpu': [
            'tensorflow==2.0.0',
        ],
        'gpu': [
            'tensorflow-gpu==2.0.0',
        ],
    },

)