Passed
Push — master ( e83024...85c52f )
by Fernando
01:01
created

LabelTransform.get_images_dict()   A

Complexity

Conditions 1

Size

Total Lines 7
Code Lines 6

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 1
eloc 6
nop 2
dl 0
loc 7
rs 10
c 0
b 0
f 0
1
from typing import List
2
3
from ....data.image import LabelMap
4
from ....data.subject import Subject
5
from ...transform import Transform
6
7
8
class LabelTransform(Transform):
9
    """Transform that modifies label maps."""
10
11
    def get_images(self, subject: Subject) -> List[LabelMap]:
12
        images = subject.get_images(
13
            intensity_only=False,
14
            include=self.include,
15
            exclude=self.exclude,
16
        )
17
        return [im for im in images if isinstance(im, LabelMap)]
18
19
    def get_images_dict(self, subject: Subject) -> List[LabelMap]:
20
        images = subject.get_images_dict(
21
            intensity_only=False,
22
            include=self.include,
23
            exclude=self.exclude,
24
        )
25
        return {k: v for (k, v) in images.items() if isinstance(v, LabelMap)}
26