Updated jax.config import

PiperOrigin-RevId: 574488049
jax-ml · Oct 18, 2023 · a79e6b1 · a79e6b1
1 parent f41e408
commit a79e6b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/triton_call_test.py b/tests/triton_call_test.py
@@ -19,7 +19,7 @@
 from absl.testing import parameterized
 import jax
 from jax import random
-from jax.config import config
+from jax import config
 import jax.numpy as jnp
 import jax_triton as jt
 import numpy as np