Replace reduce_labels with do_reduce_labels (#21218)

* Replace reduce_labels with do_reduce_labels

* Replace only for __init__ and preprocess

* Update tests
This commit is contained in:
amyeroberts
2023-01-23 17:21:33 +00:00
committed by GitHub
parent 1eda4a4102
commit 354ea44340
5 changed files with 75 additions and 29 deletions

View File

@@ -53,7 +53,7 @@ class MaskFormerFeatureExtractionTester(unittest.TestCase):
image_mean=[0.5, 0.5, 0.5],
image_std=[0.5, 0.5, 0.5],
num_labels=10,
reduce_labels=True,
do_reduce_labels=True,
ignore_index=255,
):
self.parent = parent
@@ -74,7 +74,7 @@ class MaskFormerFeatureExtractionTester(unittest.TestCase):
self.height = 3
self.width = 4
self.num_labels = num_labels
self.reduce_labels = reduce_labels
self.do_reduce_labels = do_reduce_labels
self.ignore_index = ignore_index
def prepare_feat_extract_dict(self):
@@ -86,7 +86,7 @@ class MaskFormerFeatureExtractionTester(unittest.TestCase):
"image_std": self.image_std,
"size_divisor": self.size_divisor,
"num_labels": self.num_labels,
"reduce_labels": self.reduce_labels,
"do_reduce_labels": self.do_reduce_labels,
"ignore_index": self.ignore_index,
}