"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "ba64ec07bbfcbd8f8c40afaece57daa180b5854b"
Unverified Commit a59eb349 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

fix missing "models" in pipeline test module (#17090)


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent dd16a113
...@@ -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__}")
......
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