We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 747bd62 commit 5ae7218Copy full SHA for 5ae7218
test/integration/local/test_serving.py
@@ -22,7 +22,7 @@
22
from sagemaker.pytorch import PyTorchModel
23
from sagemaker.predictor import BytesDeserializer, csv_deserializer, csv_serializer, \
24
json_deserializer, json_serializer, npy_serializer, numpy_deserializer
25
-from sagemaker_containers.beta.framework import content_types
+from sagemaker_inference import content_types
26
from torchvision import datasets, transforms
27
28
from integration import training_dir, mnist_1d_script, model_cpu_tar, mnist_cpu_script, \
0 commit comments