(weights, Y0s, sigma, mu_0t)
| 47 | |
| 48 | @jax.jit |
| 49 | def cem_update(weights, Y0s, sigma, mu_0t): |
| 50 | idx = jnp.argsort(weights)[::-1][:10] |
| 51 | mu_0tm1 = jnp.mean(Y0s[idx], axis=0) |
| 52 | return mu_0tm1, sigma |
| 53 | |
| 54 | |
| 55 | def run_path_integral(args: Args): |
nothing calls this directly
no outgoing calls
no test coverage detected