Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
ef978d0a
Unverified
Commit
ef978d0a
authored
Oct 23, 2023
by
Arthur
Committed by
GitHub
Oct 23, 2023
Browse files
skip two tests (#27013)
* skip two tests * skip torch as well * fixup
parent
45425660
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
12 additions
and
1 deletion
+12
-1
tests/models/blenderbot_small/test_modeling_blenderbot_small.py
...models/blenderbot_small/test_modeling_blenderbot_small.py
+3
-1
tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py
...els/blenderbot_small/test_modeling_tf_blenderbot_small.py
+9
-0
No files found.
tests/models/blenderbot_small/test_modeling_blenderbot_small.py
View file @
ef978d0a
...
...
@@ -244,7 +244,9 @@ class BlenderbotSmallModelTest(ModelTesterMixin, GenerationTesterMixin, Pipeline
):
if
pipeline_test_casse_name
==
"TextGenerationPipelineTests"
:
return
True
# TODO @Rocketnight1 to fix
if
pipeline_test_casse_name
==
"ConversationalPipelineTests"
:
return
True
return
False
def
setUp
(
self
):
...
...
tests/models/blenderbot_small/test_modeling_tf_blenderbot_small.py
View file @
ef978d0a
...
...
@@ -209,6 +209,15 @@ class TFBlenderbotSmallModelTest(TFModelTesterMixin, PipelineTesterMixin, unitte
config_and_inputs
=
self
.
model_tester
.
prepare_config_and_inputs_for_common
()
self
.
model_tester
.
check_decoder_model_past_large_inputs
(
*
config_and_inputs
)
# TODO: Fix the failed tests when this model gets more usage
def
is_pipeline_test_to_skip
(
self
,
pipeline_test_casse_name
,
config_class
,
model_architecture
,
tokenizer_name
,
processor_name
):
# TODO @Rocketnight1 to fix
if
pipeline_test_casse_name
==
"ConversationalPipelineTests"
:
return
True
return
False
@
require_tokenizers
@
require_tf
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment