File tree Expand file tree Collapse file tree 2 files changed +7
-1
lines changed Expand file tree Collapse file tree 2 files changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -6,6 +6,8 @@ CHANGELOG
6
6
==========
7
7
8
8
* doc-fix: Remove incorrect parameter for EI TFS Python README
9
+ * feature: ``Predictor ``: delete SageMaker model
10
+ * feature: ``Pipeline ``: delete SageMaker model
9
11
10
12
1.18.3.post1
11
13
============
Original file line number Diff line number Diff line change @@ -133,11 +133,15 @@ def delete_model(self):
133
133
for model_name in self ._model_names :
134
134
self .sagemaker_session .delete_model (model_name )
135
135
136
- def _get_model_names (self ):
136
+ def _get_endpoint_config_desc (self ):
137
137
endpoint_desc = self .sagemaker_session .sagemaker_client .describe_endpoint (EndpointName = self .endpoint )
138
138
self ._endpoint_config_name = endpoint_desc ['EndpointConfigName' ]
139
139
endpoint_config = self .sagemaker_session .sagemaker_client .describe_endpoint_config (
140
140
EndpointConfigName = self ._endpoint_config_name )
141
+ return endpoint_config
142
+
143
+ def _get_model_names (self ):
144
+ endpoint_config = self ._get_endpoint_config_desc ()
141
145
production_variants = endpoint_config ['ProductionVariants' ]
142
146
return map (lambda d : d ['ModelName' ], production_variants )
143
147
You can’t perform that action at this time.
0 commit comments