From 070e12f8863c37ee4d63f30d5098260ce062cfe1 Mon Sep 17 00:00:00 2001 From: Chester Curme Date: Mon, 3 Feb 2025 11:50:58 -0500 Subject: [PATCH] update check_diff script --- .github/scripts/check_diff.py | 23 +++++++++++------------ 1 file changed, 11 insertions(+), 12 deletions(-) diff --git a/.github/scripts/check_diff.py b/.github/scripts/check_diff.py index 644eff1bcd7..c6e20b99c30 100644 --- a/.github/scripts/check_diff.py +++ b/.github/scripts/check_diff.py @@ -61,12 +61,12 @@ def dependents_graph() -> dict: # load regular and test deps from pyproject.toml with open(path, "rb") as f: - pyproject = tomllib.load(f)["tool"]["poetry"] + pyproject = tomllib.load(f) pkg_dir = "libs" + "/".join(path.split("libs")[1].split("/")[:-1]) for dep in [ - *pyproject["dependencies"].keys(), - *pyproject["group"]["test"]["dependencies"].keys(), + *pyproject["project"]["dependencies"].keys(), + *pyproject["dependency-groups"]["test"].keys(), ]: if "langchain" in dep: dependents[dep].add(pkg_dir) @@ -120,8 +120,7 @@ def _get_configs_for_single_dir(job: str, dir_: str) -> List[Dict[str, str]]: py_versions = ["3.9", "3.10", "3.11", "3.12", "3.13"] # custom logic for specific directories elif dir_ == "libs/partners/milvus": - # milvus poetry doesn't allow 3.12 because they - # declare deps in funny way + # milvus doesn't allow 3.12 because they declare deps in funny way py_versions = ["3.9", "3.11"] elif dir_ in PY_312_MAX_PACKAGES: @@ -148,17 +147,17 @@ def _get_configs_for_single_dir(job: str, dir_: str) -> List[Dict[str, str]]: def _get_pydantic_test_configs( dir_: str, *, python_version: str = "3.11" ) -> List[Dict[str, str]]: - with open("./libs/core/poetry.lock", "rb") as f: - core_poetry_lock_data = tomllib.load(f) - for package in core_poetry_lock_data["package"]: + with open("./libs/core/uv.lock", "rb") as f: + core_uv_lock_data = tomllib.load(f) + for package in core_uv_lock_data["package"]: if package["name"] == "pydantic": core_max_pydantic_minor = package["version"].split(".")[1] break - with open(f"./{dir_}/poetry.lock", "rb") as f: - dir_poetry_lock_data = tomllib.load(f) + with open(f"./{dir_}/uv.lock", "rb") as f: + dir_uv_lock_data = tomllib.load(f) - for package in dir_poetry_lock_data["package"]: + for package in dir_uv_lock_data["package"]: if package["name"] == "pydantic": dir_max_pydantic_minor = package["version"].split(".")[1] break @@ -304,7 +303,7 @@ if __name__ == "__main__": f"Unknown lib: {file}. check_diff.py likely needs " "an update for this new library!" ) - elif file.startswith("docs/") or file in ["pyproject.toml", "poetry.lock"]: # docs or root poetry files + elif file.startswith("docs/") or file in ["pyproject.toml", "uv.lock"]: # docs or root uv files docs_edited = True dirs_to_run["lint"].add(".")