Skip to content

Commit c89b8f5

Browse files
author
Chuyang Deng
committed
Modify method name (_get_model_names).
1 parent edd00a6 commit c89b8f5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/predictor.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def __init__(self, endpoint, sagemaker_session=None, serializer=None, deserializ
5757
self.content_type = content_type or getattr(serializer, 'content_type', None)
5858
self.accept = accept or getattr(deserializer, 'accept', None)
5959
self._endpoint_config_name = self._get_endpoint_config_name()
60-
self._model_names = self._endpoint_config_desc_and_model_names()
60+
self._model_names = self._get_model_names()
6161

6262
def predict(self, data, initial_args=None):
6363
"""Return the inference from the specified endpoint.
@@ -140,7 +140,7 @@ def _get_endpoint_config_name(self):
140140
endpoint_config_name = endpoint_desc['EndpointConfigName']
141141
return endpoint_config_name
142142

143-
def _endpoint_config_desc_and_model_names(self):
143+
def _get_model_names(self):
144144
endpoint_config = self.sagemaker_session.sagemaker_client.describe_endpoint_config(
145145
EndpointConfigName=self._endpoint_config_name)
146146
production_variants = endpoint_config['ProductionVariants']

0 commit comments

Comments
 (0)