From 906e1d047c0332985b14b5c508dd3aa632214d3c Mon Sep 17 00:00:00 2001
From: Jakob Roth <roth@mpa-garching.mpg.de>
Date: Fri, 21 Mar 2025 11:45:08 +0100
Subject: [PATCH] test_optimize_kl: use non-static ncg
See: #426
---
test/test_re/test_optimize_kl.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/test/test_re/test_optimize_kl.py b/test/test_re/test_optimize_kl.py
index 1e5f8a288..5168c13e4 100644
--- a/test/test_re/test_optimize_kl.py
+++ b/test/test_re/test_optimize_kl.py
@@ -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,
--
GitLab