Skip to content

Commit

Permalink
fix: Access jax.config from jax
Browse files Browse the repository at this point in the history
* As of jax and jaxlib v0.4.20 accessing jax.config from the jax.config
  submodule is deprecated and it should be accessed from the jax top
  level API instead.

   ```
   >>> from jax.config import config
   <stdin>:1: DeprecationWarning: Accessing jax.config via the jax.config submodule is deprecated
   ```
  • Loading branch information
matthewfeickert committed Nov 4, 2023
1 parent aa22be9 commit f029794
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/pyhf/tensor/jax_backend.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from jax.config import config
from jax import config

config.update('jax_enable_x64', True)

Expand Down

0 comments on commit f029794

Please sign in to comment.