From c94b4e261c46888f00acc3f05d8e895d69f1fa64 Mon Sep 17 00:00:00 2001
From: kourosh hakhamaneshi <31483498+kouroshHakha@users.noreply.github.com>
Date: Wed, 9 Aug 2023 11:38:21 -0700
Subject: [PATCH] [RLlib] Remove JAX from the `test_catalog` test suite.
 (#38261)

---
 rllib/tests/test_catalog.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/rllib/tests/test_catalog.py b/rllib/tests/test_catalog.py
index 15f90a0ec401b..119d1d9614e30 100644
--- a/rllib/tests/test_catalog.py
+++ b/rllib/tests/test_catalog.py
@@ -144,9 +144,6 @@ def test_default_models(self):
                 flat_complex_input_case,
                 nested_complex_input_case,
             ],
-            "jax": [
-                flat_input_case,
-            ],
         }
 
         for fw, test_cases in test_suite.items():