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

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

Browse Source 
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2022-05-05 16:12:01 +02:00
committed by GitHub
parent dd16a113a4
commit a59eb349c5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/pipelines/test_pipelines_common.py
View File

@@ -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__}")