Passed
Push — master ( 199200...2b2f7b )
by Simon
04:29
created

hyperactive.optimizers.strategies.custom_optimization_strategy   A

Complexity

Total Complexity 2

Size/Duplication

Total Lines 22
Duplicated Lines 0 %

Importance

Changes 0
Metric Value
eloc 13
dl 0
loc 22
rs 10
c 0
b 0
f 0
wmc 2

2 Methods

Rating   Name   Duplication   Size   Complexity  
A CustomOptimizationStrategy.__init__() 0 5 1
A CustomOptimizationStrategy.add_optimizer() 0 7 1
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