diff --git a/flytekit/lazy_import/lazy_module.py b/flytekit/lazy_import/lazy_module.py index 09d159f13d..2c72be970a 100644 --- a/flytekit/lazy_import/lazy_module.py +++ b/flytekit/lazy_import/lazy_module.py @@ -11,7 +11,7 @@ def __init__(self, module_name: str): self._module_name = module_name def __getattribute__(self, attr): - raise ImportError(f"Module is not yet installed.") + raise ImportError(f"Module {object.__getattribute__(self, '_module_name')} is not yet installed.") def is_imported(module_name):