From cf2226a8c29763af568d1a4a272e9462607a50c1 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Wed, 4 May 2022 21:27:16 +0200 Subject: [PATCH] fix missing "models" in pipeline test module --- tests/pipelines/test_pipelines_common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/pipelines/test_pipelines_common.py b/tests/pipelines/test_pipelines_common.py index fe0dbd28153b..818191b72518 100644 --- a/tests/pipelines/test_pipelines_common.py +++ b/tests/pipelines/test_pipelines_common.py @@ -65,7 +65,7 @@ def get_tiny_config_from_class(configuration_class): try: 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) except (ImportError, AttributeError): logger.error(f"No model tester class for {configuration_class.__name__}")