Skip to content

Commit 47f913e

Browse files
committed
Update integ test to get the model name with method
1 parent be5c379 commit 47f913e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/integ/test_mxnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def _deploy_estimator_and_assert_instance_type(estimator, instance_type):
9797
try:
9898
predictor = estimator.deploy(1, instance_type)
9999

100-
model_name = predictor._model_names[0]
100+
model_name = predictor._get_model_names()[0]
101101
config_name = sagemaker_session.sagemaker_client.describe_endpoint(
102102
EndpointName=predictor.endpoint_name
103103
)["EndpointConfigName"]

0 commit comments

Comments
 (0)