fix tests

Author: sydney-runkle

libs/core/langchain_core/_api/__init__.py

@@ -53,7 +53,7 @@
     "LangChainDeprecationWarning": "deprecation",
     "deprecated": "deprecation",
     "surface_langchain_deprecation_warnings": "deprecation",
-    "suppress_langchain_deprecation_warnings": "deprecation",
+    "suppress_langchain_deprecation_warning": "deprecation",
     "warn_deprecated": "deprecation",
 }
 

libs/core/langchain_core/_lazy_imports.py

@@ -17,22 +17,15 @@ def create_dynamic_getattr(
 
     def _dynamic_getattr(attr_name: str) -> object:
         module_name = dynamic_imports.get(attr_name)
-        if module_name is None:
-            import_error_msg = (
-                f"cannot import name '{attr_name}' from '{package_name}.{module_path}'"
-            )
-            raise ImportError(import_error_msg)
 
-        if module_name == "__module__":
-            attr = import_module(f".{module_path}.{attr_name}", package=package_name)
-            globals()[attr_name] = attr
-            return attr
+        if module_name == "__module__" or module_name is None:
+            module = import_module(f".{module_path}.{attr_name}", package=package_name)
+            globals()[attr_name] = module
+            return module
 
         module = import_module(f".{module_path}.{module_name}", package=package_name)
         attr = getattr(module, attr_name)
-
-        g = globals()
-        g[attr_name] = attr
+        globals()[attr_name] = attr
 
         # TODO: determine if we want to do this eagerly
         # for k, v_module_name in dynamic_imports.items():

libs/core/tests/benchmarks/test_imports.py

@@ -22,7 +22,7 @@
         ),
         pytest.param(
             "from langchain_core.prompts import ChatPromptTemplate",
-            id="PromChatPromptTemplateptTemplate",
+            id="ChatPromptTemplate",
         ),
         pytest.param("from langchain_core.documents import Document", id="Document"),
         pytest.param(