Skip to content

Commit 983fccc

Browse files
committed
Add current_iteration option to callback args
1 parent 482ee73 commit 983fccc

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

CADETProcess/optimization/optimizationProblem.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1536,6 +1536,7 @@ def evaluate_callbacks(self, ind, current_iteration=1, force=False):
15361536
continue
15371537

15381538
callback._ind = ind
1539+
callback._current_iteration = current_iteration
15391540

15401541
try:
15411542
self._evaluate(x, callback, force)
@@ -3050,7 +3051,9 @@ class Callback(metaclass=StructMeta):
30503051
Callable must implement function with the following signature:
30513052
results : obj
30523053
x or final result of evaluation toolchain.
3053-
individual : Individual, optional
3054+
_current_iteration: int
3055+
Current iteration.
3056+
_individual : Individual, optional
30543057
Information about current step of optimzer.
30553058
evaluation_object : obj, optional
30563059
Current evaluation object.
@@ -3129,6 +3132,8 @@ def __call__(self, request, evaluation_object):
31293132
kwargs = self.kwargs
31303133

31313134
signature = inspect.signature(self.callback).parameters
3135+
if 'current_iteration' in signature:
3136+
kwargs['current_iteration'] = self._current_iteration
31323137
if 'individual' in signature:
31333138
kwargs['individual'] = self._ind
31343139
if 'evaluation_object' in signature:

0 commit comments

Comments
 (0)