Extra requirements in setup.py, installed with [rl]

41c791e9 · fheinrich · d75c2a5a · 41c791e9
Commit 41c791e9 authored 1 year ago by fheinrich
--- 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",]}
 )