From e4f8d7e8cd72f237020ffb97df5e097ac8e5951e Mon Sep 17 00:00:00 2001 From: Kalyan Chakravarthy Date: Mon, 27 Jan 2025 10:32:21 +0530 Subject: [PATCH] feat: improve error handling for module imports in PretrainedModelForQA --- langtest/modelhandler/llm_modelhandler.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/langtest/modelhandler/llm_modelhandler.py b/langtest/modelhandler/llm_modelhandler.py index 006f27107..af06e4627 100644 --- a/langtest/modelhandler/llm_modelhandler.py +++ b/langtest/modelhandler/llm_modelhandler.py @@ -165,6 +165,12 @@ def load_model(cls, hub: str, path: str, *args, **kwargs) -> "PretrainedModelFor cls.model = model(**path) return cls(hub, cls.model, *args, **filtered_kwargs) + except ModuleNotFoundError: + module = CHAT_MODEL_CLASSES[hub].get("module", None) + if module: + package_name = module.split(".")[0] + raise ValueError(Errors.E078(hub=hub, lib=package_name)) + raise ValueError(Errors.E078(hub=hub, lib=package_name)) except ImportError: raise ValueError(Errors.E044(path=path)) except ValidationError as e: