Skip to content
Snippets Groups Projects
Commit 906e1d04 authored by Jakob Roth's avatar Jakob Roth Committed by Gordian Edenhofer
Browse files

test_optimize_kl: use non-static ncg

See: #426
parent 465c1141
No related branches found
No related tags found
1 merge request!976Remove jax version constraint as recent version works again
Pipeline #245315 passed
......@@ -222,7 +222,9 @@ def test_optimize_kl_sample_consistency(
point_estimates=point_estimates,
draw_linear_kwargs=draw_linear_kwargs,
nonlinearly_update_kwargs=dict(minimize_kwargs=minimize_kwargs),
kl_kwargs=dict(minimize_kwargs=dict(name="M", maxiter=0)),
kl_kwargs=dict(
minimize=jft.optimize._newton_cg, minimize_kwargs=dict(name="M", maxiter=0)
),
sample_mode=sample_mode,
kl_jit=False,
residual_jit=False,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment