"examples/vscode:/vscode.git/clone" did not exist on "e799e0f1ed09ea1dca7d92cf431ff45235ace859"
Unverified Commit 9b78b070 authored by Nicolas Patry's avatar Nicolas Patry Committed by GitHub
Browse files

Fixing tests on master. (#14317)

* Fixing tests on master.

* Better fix.

* Lxmert doesn't have feature extractor but is bimodal.
parent df1f94eb
......@@ -15,12 +15,11 @@
import unittest
from transformers import (
FEATURE_EXTRACTOR_MAPPING,
MODEL_MAPPING,
TF_MODEL_MAPPING,
CLIPConfig,
FeatureExtractionPipeline,
LxmertConfig,
Wav2Vec2Config,
pipeline,
)
from transformers.testing_utils import is_pipeline_test, nested_simplify, require_tf, require_torch
......@@ -73,11 +72,8 @@ class FeatureExtractionPipelineTests(unittest.TestCase, metaclass=PipelineTestCa
if tokenizer is None:
self.skipTest("No tokenizer")
return
elif isinstance(model.config, (LxmertConfig, CLIPConfig, Wav2Vec2Config)):
self.skipTest(
"This is an Lxmert bimodal model, we need to find a more consistent way to switch on those models."
)
elif type(model.config) in FEATURE_EXTRACTOR_MAPPING or isinstance(model.config, LxmertConfig):
self.skipTest("This is a bimodal model, we need to find a more consistent way to switch on those models.")
return
elif model.config.is_encoder_decoder:
self.skipTest(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment