From 3d133cc5576e08a37a7f8a3a39a5ef3316c98c99 Mon Sep 17 00:00:00 2001 From: Matt Date: Wed, 2 Apr 2025 17:19:33 +0100 Subject: [PATCH] Stop DOSing the Hub in the CI (#37209) * As the title suggests, stop hammering the same files * make fixup * Use shutil instead of pathlib --- tests/models/fuyu/test_processor_fuyu.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/tests/models/fuyu/test_processor_fuyu.py b/tests/models/fuyu/test_processor_fuyu.py index 39a4729304..d3ce39fa89 100644 --- a/tests/models/fuyu/test_processor_fuyu.py +++ b/tests/models/fuyu/test_processor_fuyu.py @@ -1,6 +1,7 @@ import io import tempfile import unittest +from shutil import rmtree import requests @@ -32,18 +33,21 @@ if is_torch_available(): class FuyuProcessingTest(ProcessorTesterMixin, unittest.TestCase): processor_class = FuyuProcessor - def setUp(self): - self.tmpdirname = tempfile.mkdtemp() + @classmethod + def setUpClass(cls): + cls.tmpdirname = tempfile.mkdtemp() + # Ensure tempdir is cleaned up even if there's a failure + cls.addClassCleanup(lambda tempdir=cls.tmpdirname: rmtree(tempdir)) image_processor = FuyuImageProcessor() tokenizer = AutoTokenizer.from_pretrained("adept/fuyu-8b") processor = FuyuProcessor(image_processor=image_processor, tokenizer=tokenizer) - processor.save_pretrained(self.tmpdirname) + processor.save_pretrained(cls.tmpdirname) - self.text_prompt = "Generate a coco-style caption.\\n" + cls.text_prompt = "Generate a coco-style caption.\\n" bus_image_url = "https://huggingface.co/datasets/hf-internal-testing/fixtures-captioning/resolve/main/bus.png" - self.bus_image_pil = Image.open(io.BytesIO(requests.get(bus_image_url).content)) + cls.bus_image_pil = Image.open(io.BytesIO(requests.get(bus_image_url).content)) def get_processor(self): image_processor = FuyuImageProcessor()