bug fix for pints optimiser

pints doesn't like popSize=None pints-team/pints#1690
CardiacModelling · Nov 14, 2024 · f869e99 · f869e99
1 parent 333659a
commit f869e99
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ionbench/optimisers/pints_optimisers/cmaes_pints.py b/ionbench/optimisers/pints_optimisers/cmaes_pints.py
@@ -28,7 +28,8 @@ def run(bm, x0=None, popSize=None, maxIter=1000, debug=False):
 
     """
     model, opt = classes_pints.pints_setup(bm, x0, pints.CMAES, maxIter, debug)
-    opt.optimiser().set_population_size(popSize)
+    if popSize is not None:
+        opt.optimiser().set_population_size(popSize)
     x, f = opt.run()
     if opt.iterations() >= maxIter:
         model.bm.set_max_iter_flag()