Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Test PR] Fix model_ref_arn #132

Open
wants to merge 3 commits into
base: master
Choose a base branch
from
Open

[Test PR] Fix model_ref_arn #132

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
f60393f
fix model_ref_arn
Jul 12, 2024
0a34f1d
lint fix
Jul 15, 2024
bfeffb9
lint
Jul 15, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions src/autogluon/cloud/backend/sagemaker_backend.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -373,6 +373,7 @@ def deploy(
wait: bool = True,
model_kwargs: Optional[Dict] = None,
deploy_kwargs: Optional[Dict] = None,
model_reference_arn: Optional[str] = None,
) -> None:
"""
Deploy a predictor as a SageMaker endpoint, which can be used to do real-time inference later.
Expand Down Expand Up @@ -497,6 +498,7 @@ def deploy(
entry_point=entry_point,
predictor_cls=predictor_cls,
env=model_kwargs_env,
model_reference_arn=model_reference_arn,
**model_kwargs,
)
if deploy_kwargs is None:
Expand Down
2 changes: 2 additions & 0 deletions src/autogluon/cloud/predictor/cloud_predictor.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -392,6 +392,7 @@ def deploy(
custom_image_uri: Optional[str] = None,
volume_size: Optional[int] = None,
wait: bool = True,
model_reference_arn: Optional[str] = None,
backend_kwargs: Optional[Dict] = None,
) -> None:
"""
Expand Down Expand Up @@ -447,6 +448,7 @@ def deploy(
custom_image_uri=custom_image_uri,
volume_size=volume_size,
wait=wait,
model_reference_arn=model_reference_arn,
**backend_kwargs,
)

Expand Down
6 changes: 6 additions & 0 deletions src/autogluon/cloud/utils/ag_sagemaker.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -129,6 +129,7 @@ def __init__(
instance_type,
custom_image_uri=None,
env=None,
model_reference_arn=None,
**kwargs,
):
image_uri = custom_image_uri
Expand All @@ -152,6 +153,7 @@ def __init__(
entry_point=entry_point,
image_uri=image_uri,
env=env,
model_reference_arn=model_reference_arn,
**kwargs,
)

Expand Down Expand Up @@ -190,6 +192,7 @@ def prepare_container_def(
accelerator_type=None,
serverless_inference_config=None,
accept_eula=None,
model_reference_arn=None,
): # pylint: disable=unused-argument
deploy_key_prefix = fw_utils.model_code_key_prefix(self.key_prefix, self.name, self.image_uri)
deploy_env = copy.deepcopy(self.env)
Expand All @@ -200,6 +203,7 @@ def prepare_container_def(
self.repacked_model_data or self.model_data,
deploy_env,
image_config=self.image_config,
model_reference_arn=self.model_reference_arn,
)


Expand All @@ -215,6 +219,7 @@ def prepare_container_def(
accelerator_type=None,
serverless_inference_config=None,
accept_eula=None,
model_reference_arn=None,
): # pylint: disable=unused-argument
deploy_env = copy.deepcopy(self.env)
deploy_env.update(self._script_mode_env_vars())
Expand All @@ -226,6 +231,7 @@ def prepare_container_def(
self.model_data,
deploy_env,
image_config=self.image_config,
model_reference_arn=self.model_reference_arn,
)


Expand Down
Loading