diff --git a/setup.py b/setup.py index 8daf673c953090b4bbf6ce910949487412e300a5..02784343b4fa9c5962a16442165a29edab0c87ca 100644 --- a/setup.py +++ b/setup.py @@ -56,4 +56,8 @@ setup( url="https://gitlab.ub.uni-bielefeld.de/scs/cocosy/overcooked-simulator", version="0.1.0", zip_safe=False, + extras_requires={"rl": ["gymnasium", + "stable-baselines3[extra]", + "opencv-python", + "wandb",]} )