diff --git a/.github/workflows/.test-machine-learning.yml b/.github/workflows/.test-machine-learning.yml index 207d182fd5..426656ffad 100644 --- a/.github/workflows/.test-machine-learning.yml +++ b/.github/workflows/.test-machine-learning.yml @@ -638,7 +638,8 @@ jobs: run: | uv sync --extra=build uv sync --extra=build --extra=compile - uv sync --extra=build --extra=compile --dev + uv sync --extra=build --extra=compile --extra=paddlepaddle-gpu + uv sync --extra=build --extra=compile --extra=paddlepaddle-gpu --dev - name: Test working-directory: machine-learning/mineru run: | diff --git a/machine-learning/mineru/pyproject.toml b/machine-learning/mineru/pyproject.toml index 3fb0951847..9ee55317d5 100644 --- a/machine-learning/mineru/pyproject.toml +++ b/machine-learning/mineru/pyproject.toml @@ -2,9 +2,6 @@ name = "mineru" version = "1.0.0" requires-python = "~=3.12.0" -dependencies = [ - "paddlepaddle-gpu==3.0.0b2", -] [project.optional-dependencies] build = [ @@ -15,6 +12,9 @@ compile = [ "detectron2", "magic-pdf[full]==1.1.0", ] +paddlepaddle-gpu = [ + "paddlepaddle-gpu==3.0.0b2", +] [dependency-groups] dev = [ diff --git a/machine-learning/mineru/uv.lock b/machine-learning/mineru/uv.lock index 134caec92a..e3eee00b67 100644 --- a/machine-learning/mineru/uv.lock +++ b/machine-learning/mineru/uv.lock @@ -1296,9 +1296,6 @@ wheels = [ name = "mineru" version = "1.0.0" source = { virtual = "." } -dependencies = [ - { name = "paddlepaddle-gpu" }, -] [package.optional-dependencies] build = [ @@ -1309,6 +1306,9 @@ compile = [ { name = "detectron2" }, { name = "magic-pdf", extra = ["full"] }, ] +paddlepaddle-gpu = [ + { name = "paddlepaddle-gpu" }, +] [package.dev-dependencies] dev = [ @@ -1321,7 +1321,7 @@ dev = [ requires-dist = [ { name = "detectron2", marker = "extra == 'compile'", git = "https://github.com/facebookresearch/detectron2.git?branch=main" }, { name = "magic-pdf", extras = ["full"], marker = "extra == 'compile'", specifier = "==1.1.0" }, - { name = "paddlepaddle-gpu", specifier = "==3.0.0b2", index = "https://www.paddlepaddle.org.cn/packages/stable/cu118" }, + { name = "paddlepaddle-gpu", marker = "extra == 'paddlepaddle-gpu'", specifier = "==3.0.0b2", index = "https://www.paddlepaddle.org.cn/packages/stable/cu118" }, { name = "setuptools", marker = "extra == 'build'" }, { name = "torch", marker = "extra == 'build'" }, ]