Fix style

azavea · May 23, 2019 · 878623a · 878623a
1 parent 63dad1d
commit 878623a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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())