Skip to content

Commit 8282434

Browse files
authored
create celery worker with inference worker profile (#327)
* create celery worker with inference worker * try get aws profile
1 parent 1ecc192 commit 8282434

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

  • model-engine/model_engine_server/core/celery

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -504,7 +504,8 @@ def _get_backend_url_and_conf(
504504
elif backend_protocol == "s3":
505505
backend_url = "s3://"
506506
if aws_role is None:
507-
aws_session = session(infra_config().profile_ml_worker)
507+
aws_profile = os.getenv("AWS_PROFILE", infra_config().profile_ml_worker)
508+
aws_session = session(aws_profile)
508509
else:
509510
aws_session = session(aws_role)
510511
out_conf_changes.update(

0 commit comments

Comments
 (0)