Completed
Pull Request — master (#43)
by Wojtek
07:29
created

Optimizer.__init__()   A

Complexity

Conditions 1

Size

Total Lines 4

Duplication

Lines 0
Ratio 0 %

Code Coverage

Tests 4
CRAP Score 1

Importance

Changes 2
Bugs 0 Features 0
Metric Value
c 2
b 0
f 0
dl 0
loc 4
ccs 4
cts 4
cp 1
rs 10
cc 1
crap 1
1
"""Basic optimizer."""
2 1
3
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 1
        """Optimize process.
13 1
        Raises:
14 1
            AttributeError: When project has incorrect structure.
15
16 1
        Returns:
17
            bool: True if success, False otherwise."""
18 1
        for stage in self.ordered_stages:
19
            if not self.process.predecessors(stage):
20 1
                self.run_pso(stage)
21
            elif len(self.process.predecessors(stage)) == 1:
22 1
                predecessor = self.process.predecessors(stage)[0]
23
                stage.input_vector = predecessor.get_output_of_stage()
24 1
                self.run_pso(stage)
25
            else:
26
                raise AttributeError('Incorrect process structure.')
27
28
            if stage.optimization_status != OptimizationStatus.success:
29
                return False
30
        return True
31 1
32 1
    def run_pso(self, stage):
33 1
        """Run pso with predefined parameters."""
34
        pso(stage, swarmsize=self.swarm_size)
35