Skip to content

Commit

Permalink
rebase on main, fix imports in test_processing_common
Browse files Browse the repository at this point in the history
  • Loading branch information
yonigozlan committed Sep 16, 2024
1 parent f799eea commit 7c504d1
Showing 1 changed file with 11 additions and 6 deletions.
17 changes: 11 additions & 6 deletions tests/test_processing_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,26 +18,31 @@
import json
import tempfile


try:
from typing import Unpack
except ImportError:
from typing_extensions import Unpack

import numpy as np

from transformers import CLIPTokenizerFast, ProcessorMixin
from transformers.models.auto.processing_auto import processor_class_from_name
from transformers.testing_utils import (
check_json_file_has_correct_format,
require_tokenizers,
require_torch,
require_vision,
)
from transformers.utils import is_vision_available


try:
from typing import Unpack
except ImportError:
from typing_extensions import Unpack
import unittest


if is_vision_available():
from PIL import Image

from transformers import CLIPImageProcessor


def prepare_image_inputs():
"""This function prepares a list of PIL images"""
Expand Down

0 comments on commit 7c504d1

Please sign in to comment.