somehow prior import got nuked

kazewong · Dec 5, 2023 · e1de1a8 · e1de1a8
1 parent c5ebc38
commit e1de1a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/jimgw/prior.py b/src/jimgw/prior.py
@@ -1,7 +1,7 @@
 import jax
 import jax.numpy as jnp
 from flowMC.nfmodel.base import Distribution
-from jaxtyping import Array, Float, Int
+from jaxtyping import Array, Float, Int, PRNGKeyArray
 from typing import Callable, Union
 from dataclasses import field