From cf886b2b878bc224178c8a03f8b2ac3dd4048cc9 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Mon, 18 Sep 2023 19:42:58 +0200
Subject: [PATCH] New optional dependency: jax (Required, so that we can make
 all gymnasium envs during our tests)

---
 setup.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.py b/setup.py
index af4ff5d..524f2d1 100644
--- 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