b2bbf49015
* Wait until budget check is complete * Ensure that multi-GPU tests run for the master branch * Fix
16 lines
415 B
Bash
Executable File
16 lines
415 B
Bash
Executable File
#!/bin/bash
|
|
|
|
set -euo pipefail
|
|
|
|
echo "--- Enforce daily budget"
|
|
|
|
source tests/buildkite/conftest.sh
|
|
|
|
if [[ $enforce_daily_budget == 0 ]]
|
|
then
|
|
echo "Automatically approving all test jobs for trunk branches"
|
|
else
|
|
aws lambda invoke --function-name XGBoostCICostWatcher --invocation-type RequestResponse --region us-west-2 response.json
|
|
python3 tests/buildkite/enforce_daily_budget.py --response response.json
|
|
fi
|