Files
ray-project--ray/ci/test_compile_llm_requirements.sh
2026-07-13 13:17:40 +08:00

47 lines
1.4 KiB
Bash
Executable File

#!/bin/bash
set -e
# Create a temporary directory for backup files and setup cleanup trap
TEMP_DIR=$(mktemp -d)
cleanup() {
echo "Cleaning up temporary directory: $TEMP_DIR"
rm -rf "$TEMP_DIR"
}
trap cleanup EXIT
echo "Created temporary directory: $TEMP_DIR"
# Create backup copies of req files to reference to
LOCK_TYPES=(rayllm_test ray_test ray rayllm)
VARIANTS=(cpu cu130)
for LOCK_TYPE in "${LOCK_TYPES[@]}"; do
for VARIANT in "${VARIANTS[@]}"; do
cp ./python/deplocks/llm/"${LOCK_TYPE}"_py312_"${VARIANT}".lock "$TEMP_DIR/${LOCK_TYPE}_py312_${VARIANT}_backup.lock"
done
done
bazel run //ci/raydepsets:raydepsets -- build ci/raydepsets/configs/rayllm.depsets.yaml
# Copy files to artifact mount on Buildkite
for LOCK_TYPE in "${LOCK_TYPES[@]}"; do
for VARIANT in "${VARIANTS[@]}"; do
cp ./python/deplocks/llm/"${LOCK_TYPE}"_py312_"${VARIANT}".lock /artifact-mount/
done
done
# Check all files and print if files are not up to date
FAILED=0
for LOCK_TYPE in "${LOCK_TYPES[@]}"; do
for VARIANT in "${VARIANTS[@]}"; do
diff -u ./python/deplocks/llm/"${LOCK_TYPE}"_py312_"${VARIANT}".lock "$TEMP_DIR/${LOCK_TYPE}_py312_${VARIANT}_backup.lock" || {
echo "${LOCK_TYPE}_py312_${VARIANT}.lock is not up to date. Please download it from Artifacts tab and git push the changes."
FAILED=1
}
done
done
if [[ $FAILED -eq 1 ]]; then
exit 1
fi