@@ -231,55 +231,6 @@ def test_register_model_package(
231
231
sagemaker_session .sagemaker_client .delete_model_package (ModelPackageName = model_package_name )
232
232
233
233
234
- def test_register_model_package_via_group (
235
- mxnet_training_job ,
236
- sagemaker_session ,
237
- mxnet_inference_latest_version ,
238
- mxnet_inference_latest_py_version ,
239
- cpu_instance_type ,
240
- ):
241
- endpoint_name = "test-mxnet-deploy-model-{}" .format (sagemaker_timestamp ())
242
- with timeout_and_delete_endpoint_by_name (endpoint_name , sagemaker_session ):
243
- desc = sagemaker_session .sagemaker_client .describe_training_job (
244
- TrainingJobName = mxnet_training_job
245
- )
246
- model_data = desc ["ModelArtifacts" ]["S3ModelArtifacts" ]
247
- script_path = os .path .join (DATA_DIR , "mxnet_mnist" , "mnist.py" )
248
- model = MXNetModel (
249
- model_data ,
250
- "SageMakerRole" ,
251
- entry_point = script_path ,
252
- py_version = mxnet_inference_latest_py_version ,
253
- sagemaker_session = sagemaker_session ,
254
- framework_version = mxnet_inference_latest_version ,
255
- )
256
- model_package_group_name = "register-model-package-{}" .format (
257
- sagemaker_timestamp ()
258
- )
259
- model_pkg = model .register (
260
- content_types = ["application/json" ],
261
- response_types = ["application/json" ],
262
- inference_instances = ["ml.m5.large" ],
263
- transform_instances = ["ml.m5.large" ],
264
- model_package_group_name = model_package_group_name ,
265
- )
266
- assert isinstance (model_pkg , ModelPackage )
267
- predictor = model .deploy (1 , cpu_instance_type , endpoint_name = endpoint_name )
268
- data = numpy .zeros (shape = (1 , 1 , 28 , 28 ))
269
- result = predictor .predict (data )
270
- assert result is not None
271
- model_packages = sagemaker_session .sagemaker_client .list_model_packages (
272
- ModelPackageGroupName = model_package_group_name
273
- )["ModelPackageSummaryList" ]
274
- for model_package in model_packages :
275
- sagemaker_session .sagemaker_client .delete_model_package (
276
- ModelPackageName = model_package ["ModelPackageArn" ]
277
- )
278
- sagemaker_session .sagemaker_client .delete_model_package_group (
279
- ModelPackageGroupName = model_package_group_name
280
- )
281
-
282
-
283
234
def test_register_model_package_versioned (
284
235
mxnet_training_job ,
285
236
sagemaker_session ,
0 commit comments