Skip to content

Commit 6ef0b2d

Browse files
committed
Fix syntax
1 parent eb9681e commit 6ef0b2d

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

pymc3/sampling_jax.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ def sample_tfp_nuts(
3939
):
4040
from tensorflow_probability.substrates import jax as tfp
4141
import jax
42+
4243
model = modelcontext(model)
4344

4445
seed = jax.random.PRNGKey(random_seed)

0 commit comments

Comments
 (0)