From 098ad0536033da3aa5a6ea664bb31029404bf1a8 Mon Sep 17 00:00:00 2001 From: ram-from-tvl Date: Thu, 25 Sep 2025 23:40:10 +0530 Subject: [PATCH] Remove lingvo from v1 pax extra to avoid install failures on Colab --- v1/pyproject.toml | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/v1/pyproject.toml b/v1/pyproject.toml index 6280f41..f45b597 100644 --- a/v1/pyproject.toml +++ b/v1/pyproject.toml @@ -42,16 +42,18 @@ absl-py = ">=1.4.0" safetensors = "^0.5.3" [tool.poetry.extras] -pax = ["paxml", "lingvo", "jax", "jaxlib"] +# Note: `lingvo` is an optional Google-internal dependency with strict Python +# version and packaging constraints that cause install failures on some +# environments (Colab etc.). We omit it from the pax extra here so users can +# opt-in explicitly if they need it and have a compatible environment. +pax = ["paxml", "jax", "jaxlib"] torch = ["torch"] [tool.poetry.dependencies.paxml] version = ">=1.4.0" python = ">=3.10,<3.11" -[tool.poetry.dependencies.lingvo] -version = ">=0.12.7" -python = ">=3.10,<3.11" + [tool.poetry.dependencies.jax] version = ">=0.4.26"