x

libs/langchain/langchain/retrievers/pupmed.py

@@ -1,7 +1,5 @@
 from typing import TYPE_CHECKING, Any
 
-from langchain_community.retrievers import PubMedRetriever
-
 from langchain._api import create_importer
 
 if TYPE_CHECKING:

libs/langchain/tests/unit_tests/test_imports.py

@@ -42,7 +42,10 @@ def test_import_all_using_dir() -> None:
             # Without init
             module_name = module_name.rsplit(".", 1)[0]
 
-        mod = importlib.import_module(module_name)
+        try:
+            mod = importlib.import_module(module_name)
+        except ModuleNotFoundError as e:
+            raise ModuleNotFoundError(f"Could not import {module_name}") from e
         all = dir(mod)
 
         for name in all: