jax.config import

theislab · Mar 8, 2024 · 0b82c2b · 0b82c2b
1 parent ff7fbc8
commit 0b82c2b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -6,7 +6,7 @@
 import jax.numpy as jnp
 import numpy as np
 import pandas as pd
-from jax.config import config
+from jax import config
 from scipy.sparse import csr_matrix
 
 import matplotlib.pyplot as plt