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
a59eb349
Unverified
Commit
a59eb349
authored
May 05, 2022
by
Yih-Dar
Committed by
GitHub
May 05, 2022
Browse files
fix missing "models" in pipeline test module (#17090)
Co-authored-by:
ydshieh
<
ydshieh@users.noreply.github.com
>
parent
dd16a113
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
tests/pipelines/test_pipelines_common.py
tests/pipelines/test_pipelines_common.py
+1
-1
No files found.
tests/pipelines/test_pipelines_common.py
View file @
a59eb349
...
@@ -65,7 +65,7 @@ def get_tiny_config_from_class(configuration_class):
...
@@ -65,7 +65,7 @@ def get_tiny_config_from_class(configuration_class):
try
:
try
:
model_slug
=
model_type
.
replace
(
"-"
,
"_"
)
model_slug
=
model_type
.
replace
(
"-"
,
"_"
)
module
=
importlib
.
import_module
(
f
".test_modeling_
{
model_slug
}
"
,
package
=
f
"tests.
{
model_slug
}
"
)
module
=
importlib
.
import_module
(
f
".test_modeling_
{
model_slug
}
"
,
package
=
f
"tests.
models.
{
model_slug
}
"
)
model_tester_class
=
getattr
(
module
,
f
"
{
camel_case_model_name
}
ModelTester"
,
None
)
model_tester_class
=
getattr
(
module
,
f
"
{
camel_case_model_name
}
ModelTester"
,
None
)
except
(
ImportError
,
AttributeError
):
except
(
ImportError
,
AttributeError
):
logger
.
error
(
f
"No model tester class for
{
configuration_class
.
__name__
}
"
)
logger
.
error
(
f
"No model tester class for
{
configuration_class
.
__name__
}
"
)
...
...
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