Skip to content

Commit b5e4daf

Browse files
authored
fix: Use env AWS_REGION in sqs_client or default to us-west-2 (#563)
1 parent c0cea60 commit b5e4daf

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

model-engine/model_engine_server/core/celery/celery_autoscaler.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,7 @@ def excluded_namespaces():
5454

5555
autoscaler_broker = os.environ.get("BROKER_NAME", SQS_BROKER)
5656
aws_profile = os.environ.get("AWS_PROFILE")
57+
aws_region = os.environ.get("AWS_REGION", "us-west-2")
5758

5859

5960
@dataclasses.dataclass
@@ -403,7 +404,7 @@ async def get_broker_metrics(
403404
class SQSBroker(AutoscalerBroker):
404405
@staticmethod
405406
def _get_sqs_queue_size(queue_name: str):
406-
sqs_client = session(aws_profile).client("sqs", region_name="us-west-2")
407+
sqs_client = session(aws_profile).client("sqs", region_name=aws_region)
407408
try:
408409
total_start_time = time.time()
409410
queue_size_hist = []

0 commit comments

Comments
 (0)