import setuptools

setuptools.setup(
    name="unilm",        # your choice, but usually same as the package 
    version="0.1.0",          # obligatory
    packages=["dit"],  # name of the folder relative to this file where the source code lives
    install_requires=[
        "setuptools==58.2.0",
        "timm==0.5.4",
        "Pillow",
        "blobfile",
        "mypy",
        "numpy",
        "pytest",
        "requests",
        "einops",
        "tensorboardX",
        "deepspeed==0.4.0",
        "scipy",
        "opencv-python",
        "datasets",
        "huggingface_hub",
        "ipywidgets",
        "shapely==1.8",
    ],      # your dependencies, if you have any
    entry_points = {          # this here is the magic that binds your function into a callable script
        "console_scripts": ["train=dit.train_net:entry_point"],
    }
)