Files
2026-07-13 13:17:40 +08:00

126 lines
4.4 KiB
YAML

build_arg_sets:
py310:
PYTHON_VERSION: "3.10"
PYTHON_SHORT: "310"
py312:
PYTHON_VERSION: "3.12"
PYTHON_SHORT: "312"
depsets:
- name: ci_ml_thirdparty_depset_${PYTHON_SHORT}
operation: compile
requirements:
- python/requirements/ml/py313/third_party.txt
output: python/deplocks/ci/ci_ml_thirdparty_depset_py${PYTHON_VERSION}.lock
constraints:
- /tmp/ray-deps/requirements_compiled_py3.13.txt
append_flags:
- --python-version=${PYTHON_VERSION}
- --python-platform=linux
- --unsafe-package pip
build_arg_sets:
- py310
- py312
pre_hooks:
- ci/raydepsets/pre_hooks/remove-compiled-headers.sh 3.13
- name: ci_ml_torchft_depset_${PYTHON_SHORT}
operation: expand
depsets:
- ci_ml_build_depset_${PYTHON_SHORT}
requirements:
- python/requirements/ml/py313/torchft.txt
output: python/deplocks/ci/ml-torchft-ci_depset_py${PYTHON_VERSION}.lock
constraints:
- /tmp/ray-deps/requirements_compiled_py3.13.txt
append_flags:
- --python-version=${PYTHON_VERSION}
- --python-platform=linux
- --unsafe-package ray
- --unsafe-package pip
build_arg_sets:
- py310
pre_hooks:
- ci/raydepsets/pre_hooks/remove-compiled-headers.sh 3.13
- name: ci_ml_build_depset_${PYTHON_SHORT}
operation: expand
depsets:
- ray_img_depset_${PYTHON_SHORT}
requirements:
- python/requirements/ml/py313/ml-requirements.txt
- python/requirements/ml/py313/dl-cpu-requirements.txt
- python/requirements/ml/py313/train-requirements.txt
- python/requirements/ml/py313/train-test-requirements.txt
- python/requirements/ml/py313/tune-requirements.txt
- python/requirements/ml/py313/tune-test-requirements.txt
- python/requirements/ml/py313/rllib-requirements.txt
- python/requirements/ml/py313/rllib-test-requirements.txt
output: python/deplocks/ci/ml-build-ci_depset_py${PYTHON_VERSION}.lock
constraints:
- /tmp/ray-deps/requirements_compiled_py3.13.txt
append_flags:
- --index https://download.pytorch.org/whl/cpu
- --python-version=${PYTHON_VERSION}
- --python-platform=linux
- --unsafe-package ray
- --unsafe-package pip
build_arg_sets:
- py310
- py312
- name: ci_ml_gpubuild_depset_${PYTHON_SHORT}
operation: expand
depsets:
- ray_img_depset_${PYTHON_SHORT}
requirements:
- python/requirements/ml/py313/ml-requirements.txt
- python/requirements/ml/py313/dl-gpu-requirements.txt
- python/requirements/ml/py313/train-requirements.txt
- python/requirements/ml/py313/train-test-requirements.txt
- python/requirements/ml/py313/tune-requirements.txt
- python/requirements/ml/py313/tune-test-requirements.txt
- python/requirements/ml/py313/rllib-requirements.txt
- python/requirements/ml/py313/rllib-test-requirements.txt
output: python/deplocks/ci/ml-gpubuild-ci_depset_py${PYTHON_VERSION}.lock
constraints:
- /tmp/ray-deps/requirements_compiled_py3.13.txt
append_flags:
- --index https://download.pytorch.org/whl/cu128
- --python-version=${PYTHON_VERSION}
- --python-platform=linux
- --unsafe-package ray
- --unsafe-package pip
build_arg_sets:
- py310
- py312
- name: ci_ml_lightning1gpubuild_depset_${PYTHON_SHORT}
operation: expand
depsets:
- ray_img_depset_${PYTHON_SHORT}
requirements:
- python/requirements/ml/py313/ml-requirements.txt
- python/requirements/ml/py313/dl-gpu-requirements.txt
- python/requirements/ml/py313/train-requirements.txt
- python/requirements/ml/py313/train-test-requirements.txt
- python/requirements/ml/py313/tune-requirements.txt
- /tmp/ray-deps/tune-test-requirements-no-lightning.txt
- python/requirements/ml/py313/rllib-requirements.txt
- python/requirements/ml/py313/rllib-test-requirements.txt
- python/requirements/ml/py313/lightning1.txt
output: python/deplocks/ci/ml-lightning1gpubuild-ci_depset_py${PYTHON_VERSION}.lock
constraints:
- /tmp/ray-deps/requirements_compiled_py3.13_no_ptl.txt
append_flags:
- --index https://download.pytorch.org/whl/cu128
- --python-version=${PYTHON_VERSION}
- --python-platform=linux
- --unsafe-package ray
- --unsafe-package pip
pre_hooks:
- ci/raydepsets/pre_hooks/strip-pytorch-lightning-constraint.sh
build_arg_sets:
- py310