From 41c791e94050068f73eff75ecb4e5c32ace60443 Mon Sep 17 00:00:00 2001 From: fheinrich <fheinrich@techfak.de> Date: Thu, 8 Feb 2024 11:39:59 +0100 Subject: [PATCH] Extra requirements in setup.py, installed with [rl] --- setup.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/setup.py b/setup.py index 8daf673c..02784343 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",]} ) -- GitLab