SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add class properties with warnings (#21195)

Browse Source 
* Replace reduce_labels with do_reduce_labels

* Replace only for __init__ and preprocess

* Add class properties with warnings

* Update tests
This commit is contained in:
amyeroberts
2023-01-23 18:45:27 +00:00
committed by GitHub
parent b80b2218b5
commit c18b4fbe9f
8 changed files with 59 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/beit/test_image_processing_beit.py
View File

@@ -353,7 +353,7 @@ class BeitImageProcessingTest(ImageProcessingSavingTestMixin, unittest.TestCase)
self.assertTrue(encoding["labels"].min().item() >= 0)
self.assertTrue(encoding["labels"].max().item() <= 150)
image_processing.reduce_labels = True
image_processing.do_reduce_labels = True
encoding = image_processing(image, map, return_tensors="pt")
self.assertTrue(encoding["labels"].min().item() >= 0)
self.assertTrue(encoding["labels"].max().item() <= 255)

2
tests/models/segformer/test_image_processing_segformer.py
View File

@@ -339,7 +339,7 @@ class SegformerImageProcessingTest(ImageProcessingSavingTestMixin, unittest.Test
self.assertTrue(encoding["labels"].min().item() >= 0)
self.assertTrue(encoding["labels"].max().item() <= 150)
image_processing.reduce_labels = True
image_processing.do_reduce_labels = True
encoding = image_processing(image, map, return_tensors="pt")
self.assertTrue(encoding["labels"].min().item() >= 0)
self.assertTrue(encoding["labels"].max().item() <= 255)