Additional improvements to CI (#8303)

* Wait until budget check is complete

* Ensure that multi-GPU tests run for the master branch

* Fix
This commit is contained in:
Philip Hyunsu Cho
2022-10-04 04:03:38 -07:00
committed by GitHub
parent d686bf52a6
commit b2bbf49015
6 changed files with 16 additions and 9 deletions

View File

@@ -6,9 +6,9 @@ echo "--- Enforce daily budget"
source tests/buildkite/conftest.sh
if [[ $is_release_branch == 1 ]]
if [[ $enforce_daily_budget == 0 ]]
then
echo "Automatically approving all test jobs for release branches"
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