Total Complexity | 2 |
Total Lines | 22 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | # Author: Simon Blanke |
||
2 | # Email: [email protected] |
||
3 | # License: MIT License |
||
4 | |||
5 | from .optimization_strategy import BaseOptimizationStrategy |
||
6 | |||
7 | |||
8 | class CustomOptimizationStrategy(BaseOptimizationStrategy): |
||
9 | def __init__(self): |
||
10 | super().__init__() |
||
11 | |||
12 | self.optimizer_setup_l = [] |
||
13 | self.duration_sum = 0 |
||
14 | |||
15 | def add_optimizer(self, optimizer, duration=1): |
||
16 | self.duration_sum += duration |
||
17 | optimizer_setup = { |
||
18 | "optimizer": optimizer, |
||
19 | "duration": duration, |
||
20 | } |
||
21 | self.optimizer_setup_l.append(optimizer_setup) |
||
22 |