Skip to content

Commit a1182f1

Browse files
committed
Removing redundant test
1 parent fbe676d commit a1182f1

File tree

1 file changed

+0
-49
lines changed

1 file changed

+0
-49
lines changed

tests/integ/test_mxnet.py

Lines changed: 0 additions & 49 deletions
Original file line numberDiff line numberDiff line change
@@ -231,55 +231,6 @@ def test_register_model_package(
231231
sagemaker_session.sagemaker_client.delete_model_package(ModelPackageName=model_package_name)
232232

233233

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-
283234
def test_register_model_package_versioned(
284235
mxnet_training_job,
285236
sagemaker_session,

0 commit comments

Comments
 (0)