@@ -98,53 +98,19 @@ def timeout_and_delete_model_with_transformer(transformer, sagemaker_session, se
98
98
try :
99
99
transformer .delete_model ()
100
100
LOGGER .info ('deleted SageMaker model {}' .format (transformer .model_name ))
101
- < << << << HEAD
102
101
103
102
_show_logs (transformer .model_name , 'Models' , sagemaker_session )
104
103
if no_errors :
105
104
_cleanup_logs (transformer .model_name , 'Models' , sagemaker_session )
106
- == == == =
107
- if no_errors :
108
- _cleanup_model_logs (transformer .model_name , sagemaker_session )
109
- > >> >> >> 334 a0d6 ... Modify some functions , tests and update docs .
110
105
return
111
106
except ClientError as ce :
112
107
if ce .response ['Error' ]['Code' ] == 'ValidationException' :
113
108
pass
114
109
sleep (10 )
115
110
116
111
117
- < << << << HEAD
118
112
def _show_logs (resource_name , resource_type , sagemaker_session ):
119
113
log_group = '/aws/sagemaker/{}/{}' .format (resource_type , resource_name )
120
- == == == =
121
- def _show_model_logs (model_name , sagemaker_session ):
122
- log_group = '/aws/sagemaker/Models/{}' .format (model_name )
123
- try :
124
- LOGGER .info ('cloudwatch logs for log group {}' .format (log_group ))
125
- logs = AWSLogs (log_group_name = log_group , log_stream_name = 'ALL' , start = '1d' ,
126
- aws_region = sagemaker_session .boto_session .region_name )
127
- logs .list_logs ()
128
- except Exception :
129
- LOGGER .exception ('Failure occurred while listing cloudwatch log group %s. Swallowing exception but printing '
130
- 'stacktrace for debugging.' , log_group )
131
-
132
-
133
- def _cleanup_model_logs (model_name , sagemaker_session ):
134
- log_group = '/aws/sagemaker/Models/{}' .format (model_name )
135
- try :
136
- LOGGER .info ('deleting cloudwatch log group {}:' .format (log_group ))
137
- cwl_client = sagemaker_session .boto_session .client ('logs' )
138
- cwl_client .delete_log_group (logGroupName = log_group )
139
- LOGGER .info ('deleted cloudwatch log group: {}' .format (log_group ))
140
- except Exception :
141
- LOGGER .exception ('Failure occurred while cleaning up cloudwatch log group %s. '
142
- 'Swallowing exception but printing stacktrace for debugging.' , log_group )
143
-
144
-
145
- def _show_endpoint_logs (endpoint_name , sagemaker_session ):
146
- log_group = '/aws/sagemaker/Endpoints/{}' .format (endpoint_name )
147
- > >> >> >> 334 a0d6 ... Modify some functions , tests and update docs .
148
114
try :
149
115
# print out logs before deletion for debuggability
150
116
LOGGER .info ('cloudwatch logs for log group {}:' .format (log_group ))
0 commit comments