setup.py

from setuptools import setup

setup(
    name="risk_biased",
    version="0.1",
    description="Risk biased trajectory prediction",
    authors=["Jean Mercat", "Haruki Nishimura"],
    author_emails=["jean.mercat@tri.global", "haruki.nishimura@tri.global"],
    license="MIT",
    packages=["risk_biased"],
    zip_safe=False,
    install_requires=[
        "torch>=1.12",
        "matplotlib",
        "numpy",
        "mmcv>=1.4.7",
        "pytorch-lightning",
        "pytest",
        "setuptools>=59.5.0",
        "wandb",
        "plotly",
        "scipy",
        "gradio",
        "einops",
        "pydantic",
    ],
)