From bd71684749bab7caf02c5c3ddffcac488e18f2a7 Mon Sep 17 00:00:00 2001 From: xuyuon <116078673+xuyuon@users.noreply.github.com> Date: Sat, 13 Jul 2024 13:45:06 +0800 Subject: [PATCH] Update likelihood.py --- src/jimgw/single_event/likelihood.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/jimgw/single_event/likelihood.py b/src/jimgw/single_event/likelihood.py index d519d5cb..6f2a2361 100644 --- a/src/jimgw/single_event/likelihood.py +++ b/src/jimgw/single_event/likelihood.py @@ -559,7 +559,7 @@ def y(x): rng_key, optimized_positions, summary = optimizer.optimize( jax.random.PRNGKey(12094), y, initial_position ) - best_fit = optimized_positions[summary["final_log_prob"].argmin()] + best_fit = optimized_positions[jnp.nanargmin(summary["final_log_prob"])] return prior.transform(prior.add_name(best_fit))