New optional dependency: jax

(Required, so that we can make all gymnasium envs during our tests)
2023-09-18 19:42:58 +02:00 · 2023-09-18 19:42:58 +02:00 · cf886b2b87
commit cf886b2b87
parent 659a717fbd
1 changed files with 1 additions and 0 deletions
--- a/setup.py
+++ b/setup.py
@ -10,6 +10,7 @@ extras = {
    'metaworld': ['metaworld @ git+https://github.com/Farama-Foundation/Metaworld.git@d155d0051630bb365ea6a824e02c66c068947439#egg=metaworld'],
    'box2d': ['gymnasium[box2d]>=0.26.0'],
    'mujoco': ['mujoco==2.3.3', 'gymnasium[mujoco]>0.26.0'],
+    'jax': ["jax >=0.4.0", "jaxlib >=0.4.0"],
 }

 # All dependencies