From d1d3530db0906ce8863658aa7988fffabe9dd6f5 Mon Sep 17 00:00:00 2001
From: VIDEAU Mathurin <mathurin.videau@inria.fr>
Date: Thu, 28 Oct 2021 22:53:00 +0000
Subject: [PATCH] Update algorithms.py
---
GPRL/algorithms.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/GPRL/algorithms.py b/GPRL/algorithms.py
index 020a652..6f110da 100644
--- a/GPRL/algorithms.py
+++ b/GPRL/algorithms.py
@@ -54,7 +54,7 @@ def eaMuPlusLambdaUCB(population, toolbox, simulation_budget, parallel_update, m
# Begin the generational process
for gen in range(1, ngen + 1):
if budget_scheduler is not None:
- simulation_budget, parallel_update = budget_scheduler(ngen, population, simulation_budget, parallel_update)
+ simulation_budget, parallel_update = budget_scheduler(gen, population, simulation_budget, parallel_update)
# Vary the population
offspring = algorithms.varOr(population, toolbox, lambda_, cxpb, mutpb)
--
GitLab