Remove repeated prepare_images in processor tests (#33163)
* Remove repeated prepare_images * Address comments - update docstring; explanatory comment
This commit is contained in:
@@ -70,6 +70,17 @@ class MgpstrProcessorTest(unittest.TestCase):
|
||||
with open(self.image_processor_file, "w", encoding="utf-8") as fp:
|
||||
json.dump(image_processor_map, fp)
|
||||
|
||||
# We copy here rather than use the ProcessorTesterMixin as this processor has a `char_tokenizer` instad of a
|
||||
# tokenizer attribute, which means all the tests would need to be overridden.
|
||||
@require_vision
|
||||
def prepare_image_inputs(self):
|
||||
"""This function prepares a list of PIL images, or a list of numpy arrays if one specifies numpify=True,
|
||||
or a list of PyTorch tensors if one specifies torchify=True.
|
||||
"""
|
||||
image_inputs = [np.random.randint(255, size=(3, 30, 400), dtype=np.uint8)]
|
||||
image_inputs = [Image.fromarray(np.moveaxis(x, 0, -1)) for x in image_inputs]
|
||||
return image_inputs
|
||||
|
||||
def get_tokenizer(self, **kwargs):
|
||||
return MgpstrTokenizer.from_pretrained(self.tmpdirname, **kwargs)
|
||||
|
||||
@@ -79,15 +90,6 @@ class MgpstrProcessorTest(unittest.TestCase):
|
||||
def tearDown(self):
|
||||
shutil.rmtree(self.tmpdirname)
|
||||
|
||||
def prepare_image_inputs(self):
|
||||
"""This function prepares a list of PIL images."""
|
||||
|
||||
image_input = np.random.randint(255, size=(3, 30, 400), dtype=np.uint8)
|
||||
|
||||
image_input = Image.fromarray(np.moveaxis(image_input, 0, -1))
|
||||
|
||||
return image_input
|
||||
|
||||
def test_save_load_pretrained_default(self):
|
||||
tokenizer = self.get_tokenizer()
|
||||
image_processor = self.get_image_processor()
|
||||
|
||||
Reference in New Issue
Block a user