Skip to content

Commit

Permalink
Fix syntax
Browse files Browse the repository at this point in the history
  • Loading branch information
twiecki committed Dec 3, 2020
1 parent eb9681e commit 6ef0b2d
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions pymc3/sampling_jax.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@ def sample_tfp_nuts(
):
from tensorflow_probability.substrates import jax as tfp
import jax

model = modelcontext(model)

seed = jax.random.PRNGKey(random_seed)
Expand Down

0 comments on commit 6ef0b2d

Please sign in to comment.