Fix code quality again

huggingface · Nov 17, 2021 · ed14f3e · ed14f3e
1 parent 4126a6a
commit ed14f3e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/transformers/models/beit/feature_extraction_beit.py b/src/transformers/models/beit/feature_extraction_beit.py
@@ -65,9 +65,9 @@ class BeitFeatureExtractor(FeatureExtractionMixin, ImageFeatureExtractionMixin):
         image_std (:obj:`List[int]`, defaults to :obj:`[0.5, 0.5, 0.5]`):
             The sequence of standard deviations for each channel, to be used when normalizing images.
         reduce_labels (:obj:`bool`, `optional`, defaults to :obj:`False`):
-            Whether or not to reduce all label values of segmentation maps by 1, and replace the labels which are 0 by
-            255. Usually used for datasets where 0 is used for background, and background itself is not included in all
-            classes of a dataset (e.g. ADE20k).
+            Whether or not to reduce all label values of segmentation maps by 1. Usually used for datasets where 0 is
+            used for background, and background itself is not included in all classes of a dataset (e.g. ADE20k). The
+            background label will be replaced by 255.
     """
 
     model_input_names = ["pixel_values"]

diff --git a/src/transformers/models/segformer/feature_extraction_segformer.py b/src/transformers/models/segformer/feature_extraction_segformer.py
@@ -60,9 +60,9 @@ class SegformerFeatureExtractor(FeatureExtractionMixin, ImageFeatureExtractionMi
             The sequence of standard deviations for each channel, to be used when normalizing images. Defaults to the
             ImageNet std.
         reduce_labels (:obj:`bool`, `optional`, defaults to :obj:`False`):
-            Whether or not to reduce all label values of segmentation maps by 1, and replace the labels which are 0 by
-            255. Usually used for datasets where 0 is used for background, and background itself is not included in all
-            classes of a dataset (e.g. ADE20k).
+            Whether or not to reduce all label values of segmentation maps by 1. Usually used for datasets where 0 is
+            used for background, and background itself is not included in all classes of a dataset (e.g. ADE20k). The
+            background label will be replaced by 255.
     """
 
     model_input_names = ["pixel_values"]