github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-27 06:18:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

x

Browse Source
This commit is contained in:
Eugene Yurtsev
2024-09-13 13:05:17 -04:00
parent f16da0c9d3
commit e4727bf17d
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
libs/community/tests/unit_tests/load/test_serializable.py
View File

@@ -45,6 +45,12 @@ def import_all_modules(package_name: str) -> dict:
for importer, modname, ispkg in pkgutil.walk_packages(
package.__path__, package.__name__ + "."
):
if modname.startswith("langchain_core.pydantic_v1") or modname.startswith(
"langchain.pydantic_v1"
):
# These are deprecated and should not be imported as they
# intentionally raise an import error
continue
try:
module = importlib.import_module(modname)
except ModuleNotFoundError:
@@ -138,8 +144,15 @@ def test_serializable_mapping() -> None:
for k, import_path in serializable_modules.items():
import_dir, import_obj = import_path[:-1], import_path[-1]
module = ".".join(import_dir)
if module.startswith("langchain_core.pydantic_v1") or module.startswith(
"langchain.pydantic_v1"
):
# These are deprecated and should not be imported as they
# intentionally raise an import error
continue
# Import module
mod = importlib.import_module(".".join(import_dir))
mod = importlib.import_module(module)
# Import class
cls = getattr(mod, import_obj)
assert list(k) == cls.lc_id()