Skip to content

Commit

Permalink
Inline arguments and sample less
Browse files Browse the repository at this point in the history
Co-authored-by: Ricardo Vieira <28983449+ricardoV94@users.noreply.github.com>
  • Loading branch information
zaxtax and ricardoV94 authored Jan 20, 2025
1 parent 81eef66 commit 4a0aa08
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tests/sampling/test_forward.py
Original file line number Diff line number Diff line change
Expand Up @@ -562,15 +562,15 @@ def test_external_trace(self):
assert list(ppc.keys()) == ["a"]

def test_external_trace_det(self):
nchains = 2
ndraws = 500
with pm.Model() as model:
mu = pm.Normal("mu", 0.0, 1.0)
a = pm.Normal("a", mu=mu, sigma=1, observed=0.0)
b = pm.Deterministic("b", a + 1)
trace = pm.sample(
draws=ndraws,
chains=nchains,
tune=50,
draws=50,
chains=1,
compute_convergence_checks=False
)

# test that trace is used in ppc
Expand Down

0 comments on commit 4a0aa08

Please sign in to comment.