fix: Access jax.config from jax (#2376)

matthewfeickert · web-flow · commit 899d1fd63158 · 2023-11-04T02:08:55.000-05:00
* 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.

   ```
   &gt;&gt;&gt; from jax.config import config
   &lt;stdin&gt;:1: DeprecationWarning: Accessing jax.config via the jax.config submodule is deprecated
   ```
diff --git a/src/pyhf/tensor/jax_backend.py b/src/pyhf/tensor/jax_backend.py
@@ -1,4 +1,4 @@
-from jax.config import config
+from jax import config
 
 config.update('jax_enable_x64', True)
 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-from jax.config import config`
	`1`	`+from jax import config`
`2`	`2`
`3`	`3`	`config.update('jax_enable_x64', True)`
`4`	`4`