Various CI savings (#8291)

2022-09-30 05:42:56 -07:00
parent 299e5000a4
commit 9af99760d4
7 changed files with 87 additions and 7 deletions
--- a/tests/buildkite/enforce_daily_budget.sh
+++ b/tests/buildkite/enforce_daily_budget.sh
@@ -0,0 +1,15 @@
+#!/bin/bash
+
+set -euo pipefail
+
+echo "--- Enforce daily budget"
+
+source tests/buildkite/conftest.sh
+
+if [[ $is_release_branch == 1 ]]
+then
+  echo "Automatically approving all test jobs for release 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