Skip to content
This repository was archived by the owner on Feb 25, 2024. It is now read-only.

Commit 49c38fa

Browse files
authored
fix: return repository_url instead of registry_url (#35)
1 parent 306c034 commit 49c38fa

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

bentoctl_sagemaker/registry_utils.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,16 +37,19 @@ def create_ecr_repository_if_not_exists(region, repository_name):
3737
return repository_id, repository_uri
3838

3939

40-
def create_repository(deployment_name, operator_spec):
40+
def create_repository(repository_name, operator_spec):
4141
"""
4242
Create ECR repository and return the information.
4343
"""
4444
repo_id, _ = create_ecr_repository_if_not_exists(
45-
operator_spec["region"], deployment_name
45+
operator_spec["region"], repository_name
4646
)
47-
repo_url, username, password = get_ecr_login_info(operator_spec["region"], repo_id)
47+
registry_url, username, password = get_ecr_login_info(
48+
operator_spec["region"], repo_id
49+
)
50+
repository_url = f"{registry_url}/{repository_name}"
4851

49-
return repo_url, username, password
52+
return repository_url, username, password
5053

5154

5255
def delete_repository(repository_name, operator_spec):

0 commit comments

Comments
 (0)