Updated jax.config import (#48)

blackjax-devs · Nov 3, 2023 · 668d6b5 · 668d6b5
1 parent 9f71185
commit 668d6b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/book/models/probabilistic_ode_solver_parameter_estimation.md b/book/models/probabilistic_ode_solver_parameter_estimation.md
@@ -102,7 +102,7 @@ plt.rcParams["axes.spines.top"] = False
 :tags: [remove-output]
 
 import jax
-from jax.config import config
+from jax import config
 
 # x64 precision
 config.update("jax_enable_x64", True)