fix lint

pyro-ppl · Sep 21, 2023 · 34c8562 · 34c8562
1 parent 8ae9eb5
commit 34c8562
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/numpyro/ops/provenance.py b/numpyro/ops/provenance.py
@@ -5,9 +5,9 @@
 from jax.api_util import flatten_fun, shaped_abstractify
 import jax.core as core
 from jax.experimental.pjit import pjit_p
+import jax.extend.linear_util as lu
 from jax.interpreters.partial_eval import trace_to_jaxpr_dynamic
 from jax.interpreters.pxla import xla_pmap_p
-import jax.extend.linear_util as lu
 import jax.numpy as jnp