added popsize kwarg in maximize_likelihood

kazewong · Nov 27, 2023 · cd01eda · cd01eda
1 parent f8c7097
commit cd01eda
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/jimgw/likelihood.py b/src/jimgw/likelihood.py
@@ -378,7 +378,7 @@ def maximize_likelihood(
         y = jax.jit(jax.vmap(y))
 
         print("Starting the optimizer")
-        optimizer = EvolutionaryOptimizer(len(bounds), verbose=True)
+        optimizer = EvolutionaryOptimizer(len(bounds), popsize=set_nwalkers, verbose=True)
         state = optimizer.optimize(y, bounds, n_loops=n_loops)
         best_fit = optimizer.get_result()[0]
         return prior.add_name(best_fit, transform_name=True, transform_value=True)