From 52f2dbe9ef72b73585352e43eb937d26b0b7166f Mon Sep 17 00:00:00 2001 From: Quentin Lhoest Date: Mon, 23 Jun 2025 17:09:46 +0200 Subject: [PATCH] fix --- tests/models/nougat/test_image_processing_nougat.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/tests/models/nougat/test_image_processing_nougat.py b/tests/models/nougat/test_image_processing_nougat.py index 6ab479c043b..996860da6ed 100644 --- a/tests/models/nougat/test_image_processing_nougat.py +++ b/tests/models/nougat/test_image_processing_nougat.py @@ -16,7 +16,6 @@ import unittest import numpy as np -from datasets import load_dataset from huggingface_hub import hf_hub_download from transformers.testing_utils import require_torch, require_vision @@ -87,8 +86,15 @@ class NougatImageProcessingTester: return self.num_channels, self.size["height"], self.size["width"] def prepare_dummy_image(self): - ds = load_dataset("hf-internal-testing/fixtures_docvqa", split="test") - return ds[0]["image"].convert("RGB") + revision = "ec57bf8c8b1653a209c13f6e9ee66b12df0fc2db" + filepath = hf_hub_download( + repo_id="hf-internal-testing/fixtures_docvqa", + filename="nougat_pdf.png", + repo_type="dataset", + revision=revision, + ) + image = Image.open(filepath).convert("RGB") + return image def prepare_image_inputs(self, equal_resolution=False, numpify=False, torchify=False): return prepare_image_inputs(