Optimizer.run_pso()   A
last analyzed

Complexity

Conditions 1

Size

Total Lines 3

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 2
CRAP Score 1

Importance

Changes 0
Metric Value
c 0
b 0
f 0
dl 0
loc 3
rs 10
cc 1
ccs 2
cts 2
cp 1
crap 1
1
"""Basic optimizer."""
2
3 1
from grortir.externals.pyswarm.pso import pso
4 1
from grortir.main.model.core.optimization_status import OptimizationStatus
5 1
from grortir.main.optimizers.base_optimizer import BaseOptimizer
6
7
8 1
class Optimizer(BaseOptimizer):
9
    """Optimizer is object which optimize process."""
10
11 1
    def optimize_process(self):
12
        """Optimize process.
13
        Raises:
14
            AttributeError: When project has incorrect structure.
15
16
        Returns:
17
            bool: True if success, False otherwise."""
18 1
        for stage in self.ordered_stages:
19 1
            if not self.process.predecessors(stage):
20 1
                self.run_pso(stage)
21 1
            elif len(self.process.predecessors(stage)) == 1:
22 1
                predecessor = self.process.predecessors(stage)[0]
23 1
                stage.input_vector = predecessor.get_output_of_stage(
24
                    predecessor.input_vector, predecessor.control_params)
25 1
                self.run_pso(stage)
26
            else:
27 1
                raise AttributeError('Incorrect process structure.')
28
29 1
            if stage.optimization_status != OptimizationStatus.success:
30 1
                return False
31 1
        return True
32
33 1
    def run_pso(self, stage):
34
        """Run pso with predefined parameters."""
35
        pso(stage, swarmsize=self.swarm_size)
36