From 1aafa916d5209cd4a54bb1b54ab235240bc2696b Mon Sep 17 00:00:00 2001 From: Sergei Lebedev Date: Thu, 19 Oct 2023 09:20:46 -0700 Subject: [PATCH] Updated jax.config import PiperOrigin-RevId: 574896273 --- tests/triton_call_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/triton_call_test.py b/tests/triton_call_test.py index 71a59806..ebdf41e9 100644 --- a/tests/triton_call_test.py +++ b/tests/triton_call_test.py @@ -18,8 +18,8 @@ from absl.testing import absltest from absl.testing import parameterized import jax +from jax import config from jax import random -from jax.config import config import jax.numpy as jnp import jax_triton as jt import numpy as np