from setuptools import setup, find_packages

setup(
    name="discrete_trpl",
    version="0.1",
    packages=find_packages(include=["discrete_trpl", "discrete_trpl.*"]),
    install_requires=[
        "torch",
        "numpy",
    ],
    python_requires=">=3.6",
    extras_require={
        "cartpole": [
            "gymnasium",
            "tyro",
            "tensorboard",
            "wandb"
        ]
    },
    description="Discrete Trust Region Policy Layers",
)
