We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c0cea60 commit b5e4dafCopy full SHA for b5e4daf
model-engine/model_engine_server/core/celery/celery_autoscaler.py
@@ -54,6 +54,7 @@ def excluded_namespaces():
54
55
autoscaler_broker = os.environ.get("BROKER_NAME", SQS_BROKER)
56
aws_profile = os.environ.get("AWS_PROFILE")
57
+aws_region = os.environ.get("AWS_REGION", "us-west-2")
58
59
60
@dataclasses.dataclass
@@ -403,7 +404,7 @@ async def get_broker_metrics(
403
404
class SQSBroker(AutoscalerBroker):
405
@staticmethod
406
def _get_sqs_queue_size(queue_name: str):
- sqs_client = session(aws_profile).client("sqs", region_name="us-west-2")
407
+ sqs_client = session(aws_profile).client("sqs", region_name=aws_region)
408
try:
409
total_start_time = time.time()
410
queue_size_hist = []
0 commit comments