SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)

Browse Source 
* add more cases

* fix method not found in unittest

Signed-off-by: Lin, Fanli <fanli.lin@intel.com>

* fix more cases

* add more models

* add all

* no unittest.case

* remove for oneformer

* fix style

---------

Signed-off-by: Lin, Fanli <fanli.lin@intel.com>
This commit is contained in:
Fanli Lin
2024-12-13 21:33:45 +08:00
committed by GitHub
parent 3d213b57fe
commit bdd4201fdb
34 changed files with 34 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/audio_spectrogram_transformer/test_feature_extraction_audio_spectrogram_transformer.py
View File

@@ -50,7 +50,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None):
return values
class ASTFeatureExtractionTester(unittest.TestCase):
class ASTFeatureExtractionTester:
def __init__(
self,
parent,

2
tests/models/beit/test_image_processing_beit.py
View File

@@ -33,7 +33,7 @@ if is_vision_available():
from transformers import BeitImageProcessor
class BeitImageProcessingTester(unittest.TestCase):
class BeitImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/clap/test_feature_extraction_clap.py
View File

@@ -53,7 +53,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None):
@require_torch
@require_torchaudio
# Copied from tests.models.whisper.test_feature_extraction_whisper.WhisperFeatureExtractionTester with Whisper->Clap
class ClapFeatureExtractionTester(unittest.TestCase):
class ClapFeatureExtractionTester:
def __init__(
self,
parent,

2
tests/models/clip/test_image_processing_clip.py
View File

@@ -26,7 +26,7 @@ if is_vision_available():
from transformers import CLIPImageProcessor
class CLIPImageProcessingTester(unittest.TestCase):
class CLIPImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/clvp/test_feature_extraction_clvp.py
View File

@@ -57,7 +57,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None):
@require_torch
class ClvpFeatureExtractionTester(unittest.TestCase):
class ClvpFeatureExtractionTester:
def __init__(
self,
parent,

2
tests/models/conditional_detr/test_image_processing_conditional_detr.py
View File

@@ -35,7 +35,7 @@ if is_vision_available():
from transformers import ConditionalDetrImageProcessor
class ConditionalDetrImageProcessingTester(unittest.TestCase):
class ConditionalDetrImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/dac/test_feature_extraction_dac.py
View File

@@ -51,7 +51,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None):
@require_torch
# Copied from transformers.tests.encodec.test_feature_extraction_dac.EncodecFeatureExtractionTester with Encodec->Dac
class DacFeatureExtractionTester(unittest.TestCase):
class DacFeatureExtractionTester:
# Ignore copy
def __init__(
self,

2
tests/models/encodec/test_feature_extraction_encodec.py
View File

@@ -50,7 +50,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None):
@require_torch
class EnCodecFeatureExtractionTester(unittest.TestCase):
class EnCodecFeatureExtractionTester:
def __init__(
self,
parent,

2
tests/models/grounding_dino/test_image_processing_grounding_dino.py
View File

@@ -37,7 +37,7 @@ if is_vision_available():
from transformers import GroundingDinoImageProcessor
class GroundingDinoImageProcessingTester(unittest.TestCase):
class GroundingDinoImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/idefics/test_image_processing_idefics.py
View File

@@ -36,7 +36,7 @@ if is_vision_available():
from transformers import IdeficsImageProcessor
class IdeficsImageProcessingTester(unittest.TestCase):
class IdeficsImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/idefics2/test_image_processing_idefics2.py
View File

@@ -34,7 +34,7 @@ if is_torch_available():
import torch
class Idefics2ImageProcessingTester(unittest.TestCase):
class Idefics2ImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/llava_next/test_image_processing_llava_next.py
View File

@@ -34,7 +34,7 @@ if is_vision_available():
from transformers import LlavaNextImageProcessor
class LlavaNextImageProcessingTester(unittest.TestCase):
class LlavaNextImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/llava_next_video/test_image_processing_llava_next_video.py
View File

@@ -33,7 +33,7 @@ if is_vision_available():
from transformers import LlavaNextVideoImageProcessor
class LlavaNextVideoProcessingTester(unittest.TestCase):
class LlavaNextVideoProcessingTester:
def __init__(
self,
parent,

2
tests/models/llava_onevision/test_image_processing_llava_onevision.py
View File

@@ -33,7 +33,7 @@ if is_vision_available():
from transformers import LlavaOnevisionImageProcessor, LlavaOnevisionVideoProcessor
class LlavaOnevisionImageProcessingTester(unittest.TestCase):
class LlavaOnevisionImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/markuplm/test_feature_extraction_markuplm.py
View File

@@ -26,7 +26,7 @@ if is_bs4_available():
from transformers import MarkupLMFeatureExtractor
class MarkupLMFeatureExtractionTester(unittest.TestCase):
class MarkupLMFeatureExtractionTester:
def __init__(self, parent):
self.parent = parent

2
tests/models/mask2former/test_image_processing_mask2former.py
View File

@@ -39,7 +39,7 @@ if is_vision_available():
from PIL import Image
class Mask2FormerImageProcessingTester(unittest.TestCase):
class Mask2FormerImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/maskformer/test_image_processing_maskformer.py
View File

@@ -38,7 +38,7 @@ if is_vision_available():
from PIL import Image
class MaskFormerImageProcessingTester(unittest.TestCase):
class MaskFormerImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/musicgen_melody/test_feature_extraction_musicgen_melody.py
View File

@@ -69,7 +69,7 @@ def get_bip_bip(bip_duration=0.125, duration=0.5, sample_rate=32000):
@require_torch
@require_torchaudio
class MusicgenMelodyFeatureExtractionTester(unittest.TestCase):
class MusicgenMelodyFeatureExtractionTester:
def __init__(
self,
parent,

2
tests/models/oneformer/test_image_processing_oneformer.py
View File

@@ -39,7 +39,7 @@ if is_vision_available():
from PIL import Image
class OneFormerImageProcessorTester(unittest.TestCase):
class OneFormerImageProcessorTester:
def __init__(
self,
parent,

2
tests/models/oneformer/test_processor_oneformer.py
View File

@@ -59,7 +59,7 @@ def prepare_metadata(class_info_file, repo_path="shi-labs/oneformer_demo"):
return metadata
class OneFormerProcessorTester(unittest.TestCase):
class OneFormerProcessorTester:
def __init__(
self,
parent,

2
tests/models/pix2struct/test_image_processing_pix2struct.py
View File

@@ -34,7 +34,7 @@ if is_vision_available():
from transformers import Pix2StructImageProcessor
class Pix2StructImageProcessingTester(unittest.TestCase):
class Pix2StructImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/pixtral/test_image_processing_pixtral.py
View File

@@ -38,7 +38,7 @@ if is_vision_available():
from transformers import PixtralImageProcessorFast
class PixtralImageProcessingTester(unittest.TestCase):
class PixtralImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/pop2piano/test_feature_extraction_pop2piano.py
View File

@@ -48,7 +48,7 @@ if requirements_available:
from transformers import Pop2PianoFeatureExtractor
class Pop2PianoFeatureExtractionTester(unittest.TestCase):
class Pop2PianoFeatureExtractionTester:
def __init__(
self,
parent,

2
tests/models/qwen2_vl/test_image_processing_qwen2_vl.py
View File

@@ -34,7 +34,7 @@ if is_vision_available():
from transformers import Qwen2VLImageProcessor
class Qwen2VLImageProcessingTester(unittest.TestCase):
class Qwen2VLImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/seamless_m4t/test_feature_extraction_seamless_m4t.py
View File

@@ -52,7 +52,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None):
@require_torch
class SeamlessM4TFeatureExtractionTester(unittest.TestCase):
class SeamlessM4TFeatureExtractionTester:
def __init__(
self,
parent,

2
tests/models/segformer/test_image_processing_segformer.py
View File

@@ -33,7 +33,7 @@ if is_vision_available():
from transformers import SegformerImageProcessor
class SegformerImageProcessingTester(unittest.TestCase):
class SegformerImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/seggpt/test_image_processing_seggpt.py
View File

@@ -35,7 +35,7 @@ if is_vision_available():
from transformers import SegGptImageProcessor
class SegGptImageProcessingTester(unittest.TestCase):
class SegGptImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/speech_to_text/test_feature_extraction_speech_to_text.py
View File

@@ -48,7 +48,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None):
@require_torch
@require_torchaudio
class Speech2TextFeatureExtractionTester(unittest.TestCase):
class Speech2TextFeatureExtractionTester:
def __init__(
self,
parent,

2
tests/models/speecht5/test_feature_extraction_speecht5.py
View File

@@ -50,7 +50,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None):
@require_torch
class SpeechT5FeatureExtractionTester(unittest.TestCase):
class SpeechT5FeatureExtractionTester:
def __init__(
self,
parent,

2
tests/models/superpoint/test_image_processing_superpoint.py
View File

@@ -33,7 +33,7 @@ if is_vision_available():
from transformers import SuperPointImageProcessor
class SuperPointImageProcessingTester(unittest.TestCase):
class SuperPointImageProcessingTester:
def __init__(
self,
parent,

2
tests/models/univnet/test_feature_extraction_univnet.py
View File

@@ -50,7 +50,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None):
return values
class UnivNetFeatureExtractionTester(unittest.TestCase):
class UnivNetFeatureExtractionTester:
def __init__(
self,
parent,

2
tests/models/wav2vec2/test_feature_extraction_wav2vec2.py
View File

@@ -44,7 +44,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None):
return values
class Wav2Vec2FeatureExtractionTester(unittest.TestCase):
class Wav2Vec2FeatureExtractionTester:
def __init__(
self,
parent,

2
tests/models/whisper/test_feature_extraction_whisper.py
View File

@@ -50,7 +50,7 @@ def floats_list(shape, scale=1.0, rng=None, name=None):
return values
class WhisperFeatureExtractionTester(unittest.TestCase):
class WhisperFeatureExtractionTester:
def __init__(
self,
parent,

2
tests/models/yolos/test_image_processing_yolos.py
View File

@@ -36,7 +36,7 @@ if is_vision_available():
from transformers import YolosImageProcessor
class YolosImageProcessingTester(unittest.TestCase):
class YolosImageProcessingTester:
def __init__(
self,
parent,