from setuptools import setup, find_packages

setup(
    name="multi_agent_submission",
    version="0.1.0",
    description="Supplementary material for a multi-agent coordination algorithm.",
    packages=find_packages(),
    install_requires=[
        "torch<2.5",
        "torchvision",
        "torchaudio",
        "torch_geometric",
        "torch_scatter",
        "torch_sparse",
        "torch_cluster",
        "torch_spline_conv",
        "cvxpy",
        "cvxpylayers",
    ],
    dependency_links=[
        "git+https://github.com/facebookresearch/BenchMARL.git#egg=BenchMARL",
        "git+https://github.com/proroklab/VectorizedMultiAgentSimulator.git#egg=VMAS",
    ],
    python_requires="==3.12.*",
)
