如何通过Bazel pip_parse安装带extras的Python包？

步骤1：生成包含Extras的完整依赖锁定文件 #

首先要确保锁定文件包含ray[air,data,default]及其所有间接依赖，手动生成方式如下：

• 创建并激活虚拟环境：

  python -m venv venv
  source venv/bin/activate  # macOS/Linux
  # 或 venv\Scripts\activate  # Windows
  

• 安装带extras的指定版本ray：

  pip install ray[air,data,default]==2.1.0
  

• 生成锁定文件：

  pip freeze > python/requirements_lock.txt
  

• 更新WORKSPACE中的requirements_lock路径，指向这个新生成的锁定文件。

步骤2：在BUILD文件中正确引用带Extras的包 #

方法A：使用requirement函数的extras参数（推荐） #

若你的rules_python版本≥0.10.0，requirement函数支持extras参数，直接指定所需extras即可：

load("@pip//:requirements.bzl", "requirement")

py_library(
    name = "lib",
    srcs = glob(["*.py"]),
    deps = [
        requirement("ray", extras=["air", "data", "default"]),
    ],
)

方法B：直接引用带Extras的完整包名 #

若rules_python版本不支持extras参数，直接在requirement中传入带extras的完整包名，pip_parse会自动将非法字符转换为Bazel允许的名称：

load("@pip//:requirements.bzl", "requirement")

py_library(
    name = "lib",
    srcs = glob(["*.py"]),
    deps = [
        requirement("ray[air,data,default]"),
    ],
)

验证解决方案 #

执行bazel run :app，此时ray的data、air等模块应能正常导入，不会再出现循环导入或依赖缺失的错误。

常见问题排查 #

• 若仍报错，检查requirements_lock.txt是否包含ray[air,data,default]的所有依赖（如ray-data、ray-air等包）。
• 确认WORKSPACE中的pip_parse配置正确，指向的是生成的锁定文件而非原始requirements.txt。
• 升级rules_python到最新版本，规避旧版本兼容性问题：

  bazel run @bazel_tools//tools/build_defs/repo:downloader -- update @rules_python//:version.bzl
  

内容的提问来源于stack exchange，提问作者andrii