Update dataquality/utils/semantic_segmentation/errors.py

rungalileo · May 22, 2023 · a76220a · a76220a
1 parent b0be115
commit a76220a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dataquality/utils/semantic_segmentation/errors.py b/dataquality/utils/semantic_segmentation/errors.py
@@ -193,7 +193,7 @@ def calculate_dep_polygons_batch(
     dep score to the mean dep score
 
     Args:
-        polygons_batch (List[List[[Polygon]]): list of the gold polygons
+        polygons_batch (List[List[[Polygon]]): list of the polygons (gold or pred)
             for an image
         dep_heatmaps (np.ndarray): heatmaps of DEP scores for an image
         height (int): height of original image to resize the dep map to the correct