24 lines
656 B
Python
24 lines
656 B
Python
import shutil
|
|
import tempfile
|
|
import unittest
|
|
|
|
from transformers import Owlv2Processor
|
|
from transformers.testing_utils import require_scipy
|
|
|
|
from ...test_processing_common import ProcessorTesterMixin
|
|
|
|
|
|
@require_scipy
|
|
class Owlv2ProcessorTest(ProcessorTesterMixin, unittest.TestCase):
|
|
processor_class = Owlv2Processor
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.tmpdirname = tempfile.mkdtemp()
|
|
processor = cls.processor_class.from_pretrained("google/owlv2-base-patch16-ensemble")
|
|
processor.save_pretrained(cls.tmpdirname)
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
shutil.rmtree(cls.tmpdirname, ignore_errors=True)
|