Skip to content

Commit 5ae7218

Browse files
Remove user of sagemaker_containers (old deprecated package) in integration tests (use sagemaker_inference instead)
1 parent 747bd62 commit 5ae7218

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/integration/local/test_serving.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222
from sagemaker.pytorch import PyTorchModel
2323
from sagemaker.predictor import BytesDeserializer, csv_deserializer, csv_serializer, \
2424
json_deserializer, json_serializer, npy_serializer, numpy_deserializer
25-
from sagemaker_containers.beta.framework import content_types
25+
from sagemaker_inference import content_types
2626
from torchvision import datasets, transforms
2727

2828
from integration import training_dir, mnist_1d_script, model_cpu_tar, mnist_cpu_script, \

0 commit comments

Comments
 (0)