From a6e3b17981364a728214169b824e36300686479f Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Fri, 4 Mar 2022 09:53:00 +0100 Subject: [PATCH] Re-enabling all fast pipeline tests. (#15924) --- tests/pipelines/test_pipelines_common.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/pipelines/test_pipelines_common.py b/tests/pipelines/test_pipelines_common.py index 9dbb3d2bca..fe0dbd2815 100644 --- a/tests/pipelines/test_pipelines_common.py +++ b/tests/pipelines/test_pipelines_common.py @@ -64,7 +64,8 @@ def get_tiny_config_from_class(configuration_class): camel_case_model_name = configuration_class.__name__.split("Config")[0] try: - module = importlib.import_module(f".test_modeling_{model_type.replace('-', '_')}", package="tests") + model_slug = model_type.replace("-", "_") + module = importlib.import_module(f".test_modeling_{model_slug}", package=f"tests.{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__}")