def test_iterator():
with pm.Model() as model:
a = pm.Normal("a", shape=1)
pm.HalfNormal("b")
step1 = pm.NUTS([a])
step2 = pm.Metropolis([model.b_log__])
step = pm.CompoundStep([step1, step2])
start = {"a": 1.0, "b_log__": 2.0}
sampler = ps.ParallelSampler(10, 10, 3, 2, [2, 3, 4], [start] * 3, step, 0, False)
with sampler:
for draw in sampler:
pass
def test_iterator():
with pm.Model() as model:
a = pm.Normal('a', shape=1)
pm.HalfNormal('b')
step1 = pm.NUTS([a])
step2 = pm.Metropolis([model.b_log__])
step = pm.CompoundStep([step1, step2])
start = time.time()
start = {'a': 1., 'b_log__': 2.}
sampler = ps.ParallelSampler(10, 10, 3, 2, [2, 3, 4], [start] * 3,
step, 0, False)
with sampler:
for draw in sampler:
pass