From 878623a8a0d1c869ed1ab91cb20554de9a6108f9 Mon Sep 17 00:00:00 2001 From: Lewis Fishgold Date: Thu, 23 May 2019 12:04:09 -0400 Subject: [PATCH] Fix style --- rastervision/core/class_map.py | 1 + tests/evaluation/test_semantic_segmentation_evaluation.py | 7 +++---- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/rastervision/core/class_map.py b/rastervision/core/class_map.py index 6ba5505b3..c068d70f3 100644 --- a/rastervision/core/class_map.py +++ b/rastervision/core/class_map.py @@ -46,6 +46,7 @@ class ClassMap(object): the class 0 is reserved for use as an "ignore" class, which denotes pixels that should not be used for training the model or evaluating it. """ + def __init__(self, class_items): """Construct a new ClassMap. diff --git a/tests/evaluation/test_semantic_segmentation_evaluation.py b/tests/evaluation/test_semantic_segmentation_evaluation.py index 3335760bd..a8e07a222 100644 --- a/tests/evaluation/test_semantic_segmentation_evaluation.py +++ b/tests/evaluation/test_semantic_segmentation_evaluation.py @@ -65,10 +65,9 @@ def test_compute_ignore_class(self): pred_raster.set_raster(pred_array) pred_label_source = SemanticSegmentationLabelSource(source=pred_raster) - class_map = ClassMap([ - ClassItem(id=0, name='ignore'), - ClassItem(id=1, name='one') - ]) + class_map = ClassMap( + [ClassItem(id=0, name='ignore'), + ClassItem(id=1, name='one')]) eval = SemanticSegmentationEvaluation(class_map) eval.compute(gt_label_source.get_labels(), pred_label_source.get_labels())