Skip to content

Commit 495837d

Browse files
committed
rename one more deployment_image_uri
1 parent e3b93cb commit 495837d

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/sagemaker/session.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -2777,7 +2777,7 @@ def endpoint_from_job(
27772777
def endpoint_from_model_data(
27782778
self,
27792779
model_s3_location,
2780-
deployment_image_uri,
2780+
image_uri,
27812781
initial_instance_count,
27822782
instance_type,
27832783
name=None,
@@ -2792,7 +2792,7 @@ def endpoint_from_model_data(
27922792
27932793
Args:
27942794
model_s3_location (str): S3 URI of the model artifacts to use for the endpoint.
2795-
deployment_image_uri (str): The Docker image URI which defines the runtime code to be
2795+
image_uri (str): The Docker image URI which defines the runtime code to be
27962796
used as the entry point for accepting prediction requests.
27972797
initial_instance_count (int): Minimum number of EC2 instances to launch. The actual
27982798
number of active instances for an endpoint at any given time varies due to
@@ -2824,7 +2824,7 @@ def endpoint_from_model_data(
28242824
28252825
"""
28262826
model_environment_vars = model_environment_vars or {}
2827-
name = name or name_from_image(deployment_image_uri)
2827+
name = name or name_from_image(image_uri)
28282828
model_vpc_config = vpc_utils.sanitize(model_vpc_config)
28292829

28302830
if _deployment_entity_exists(
@@ -2838,7 +2838,7 @@ def endpoint_from_model_data(
28382838
lambda: self.sagemaker_client.describe_model(ModelName=name)
28392839
):
28402840
primary_container = container_def(
2841-
image_uri=deployment_image_uri,
2841+
image_uri=image_uri,
28422842
model_data_url=model_s3_location,
28432843
env=model_environment_vars,
28442844
)

0 commit comments

Comments
 (0)