fix import

pyro-ppl · Dec 20, 2024 · a14adaa · a14adaa
1 parent ba9a07a
commit a14adaa
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/numpyro/primitives.py b/numpyro/primitives.py
@@ -7,14 +7,13 @@
 from typing import Callable, Generator, Optional, Union, cast
 import warnings
 
-from distributions.distribution import DistributionLike
-
 import jax
 from jax import Array, lax, random
 import jax.numpy as jnp
 from jax.typing import ArrayLike
 
 import numpyro
+from numpyro.distributions.distribution import DistributionLike
 from numpyro.util import find_stack_level, identity
 
 _PYRO_STACK: list = []