diff --git a/jax/experimental/jax2tf/tests/primitives_test.py b/jax/experimental/jax2tf/tests/primitives_test.py
index <HASH>..<HASH> 100644
--- a/jax/experimental/jax2tf/tests/primitives_test.py
+++ b/jax/experimental/jax2tf/tests/primitives_test.py
@@ -81,13 +81,13 @@ class JaxPrimitiveTest(tf_test_util.JaxToTfTestCase):
       else:
         self.assertIn(p, tf_impl)
 
-  @parameterized.named_parameters(jtu.cases_from_list(
+  @parameterized.named_parameters(
     dict(testcase_name=f"_{f_jax.__name__}",
          f_jax=f_jax)
     for f_jax in [jnp.add, jnp.subtract, jnp.multiply, jnp.divide,
                   jnp.less, jnp.less_equal, jnp.equal, jnp.greater,
                   jnp.greater_equal, jnp.not_equal, jnp.maximum,
-                  jnp.minimum]))
+                  jnp.minimum])
   def test_type_promotion(self, f_jax=jnp.add):
     # We only test a few types here, as tensorflow does not support many
     # types like uint* or bool in binary ops.
[jax2tf] Build a primitive harness for test_type_promotion. (#<I>)

* [jax2tf] Build a primitive harness for test_type_promotion.



We were previously generating the cases using `jtu.cases_from_list`,

which by default dropped 2 test cases (JAX_NUM_GENERATED_CASES=<I>,

number of generated cases = <I>).



* [jax2tf] Fix the generated test cases for test_type_promotion.