from setuptools import setup, find_namespace_packages

setup(
    name='convexrobust',
    packages=find_namespace_packages(include=['convexrobust.*', 'lib.*']),
    version='0.1',
    install_requires=[
        'torch',
        'pytorch-lightning==1.6.4',
        'pytorch-lightning-bolts',
        'torchattacks',
        'torchvision',
        'numpy',
        'scipy',
        'statsmodels',
        'matplotlib',
        'click',
        'tqdm',
        'colorama',
        'sklearn',
        'gym',
        'cvxpy',
        'PyQt5',
        'lmdb',
        'baker',
        'msgpack',
        'logzero',
        'config',
        'lightgbm',
        'lief',
        'einops',
        'foolbox',
        'torchnet',
    ])
