Skip to content

Commit df03312

Browse files
authored
[Monitoring] Update auto scale worker script (Cloud-CV#4235)
* [Monitoring] Remove excluded challenges from auto scaling * Auto scale non-remote workers * Fix flake8 issue
1 parent c34f8f7 commit df03312

1 file changed

Lines changed: 4 additions & 26 deletions

File tree

scripts/monitoring/auto_scale_workers.py

Lines changed: 4 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -10,28 +10,6 @@
1010
warnings.filterwarnings("ignore")
1111

1212
utc = pytz.UTC
13-
PROD_EXCLUDED_CHALLENGE_QUEUES = [
14-
"textvqa-challenge-2021-874-production-41966973-4d99-4326-a402-b749b1d89aad",
15-
"vqa-challenge-2021-830-production-6343db53-af82-4618-8c51-0e294611315a",
16-
"ego4d-state-change-object-detection-challenge-1632",
17-
"argoverse-3d-tracking-competition-52b36364-110d-47f1-8fa7-8873aa2d9965",
18-
"argoverse-motion-forecasting-competition-81d1a3c6-f7b3-4830-9dcf-3ccc9f29d3a6",
19-
"argoverse-3d-detection-competition-725-71837a72-afb0-4403-9cfc-bb3e80d733ab",
20-
"argoverse-stereo-competition-917-production-884becea-a882-43bd-8ac7-0561bd705c4e",
21-
"nuscenes-prediction-challenge-448a7b5e-6b08-4587-9293-ac07a530b426",
22-
"nuscenes-detection-challenge-510c8c6d-a0d2-40bd-95dd-b7c8ea593d03",
23-
"gqa-real-world-visual-reasoning-challenge-2019-2e1c901b-5d86-4516-b82d-8d98",
24-
"nocaps-xd-9f4bead9-b3d6-4207-8e68-a6807d786c3c",
25-
"nocaps-18b403b5-8946-4319-9fc1-758981f7f724",
26-
"nuscenes-tracking-challenge-b1ea8e46-cee9-4591-856a-7c31947d74ed",
27-
"ego4d-poc-your-challenge-name--1637-production-bfb81565-71fd-4b7b-949d-2560dd523",
28-
"ego4d-poc-your-challenge-name--1624-production-6e1419b9-f909-4428-951d-6e927676f",
29-
"dialoglue-708-d2225333-1ade-41c5-bb0c-c02c15d43e05",
30-
"vizwiz-caption-challenge-2021-739-production-a6420029-bf1c-4339-9c55-36a8913cca8",
31-
"textcaps-challenge-2020-906-production-fcc74455-ce3e-4622-b6fe-0340bcc3d228",
32-
"vizwiz-vqa-challenge-2021-743-production-862e0cf8-0611-4aa5-a05a-14e07576a513",
33-
"cvpr2023-bdd100k-multiple-object-tracking-challeng-1989-production-cdba4595-fb13",
34-
]
3513

3614
ENV = os.environ.get("ENV", "dev")
3715

@@ -113,11 +91,11 @@ def scale_up_or_down_workers(challenge, challenge_metrics):
11391
def scale_up_or_down_workers_for_challenges(response, metrics):
11492
for challenge in response["results"]:
11593
if ENV == "prod":
116-
if challenge["queue"] not in PROD_EXCLUDED_CHALLENGE_QUEUES:
117-
try:
94+
try:
95+
if challenge["remote_evaluation"] is False:
11896
scale_up_or_down_workers(challenge, metrics[str(challenge["id"])])
119-
except Exception as e:
120-
print(e)
97+
except Exception as e:
98+
print(e)
12199
else:
122100
try:
123101
scale_up_or_down_workers(challenge, metrics[str(challenge["id"])])

0 commit comments

Comments
 (0)