Skip to content

Commit 6d84df9

Browse files
author
Balaji Veeramani
committed
Update test_multi_variant_endpoint.py
1 parent 7c0cf42 commit 6d84df9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integ/test_multi_variant_endpoint.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import scipy.stats as st
2121

2222
from sagemaker import image_uris
23+
from sagemaker.deserializers import CSVDeserializer
2324
from sagemaker.s3 import S3Uploader
2425
from sagemaker.session import production_variant
2526
from sagemaker.sparkml import SparkMLModel
@@ -299,8 +300,7 @@ def test_predict_invocation_with_target_variant_local_mode(
299300
endpoint_name=multi_variant_endpoint.endpoint_name,
300301
sagemaker_session=sagemaker_session,
301302
serializer=CSVSerializer(),
302-
content_type="text/csv",
303-
accept="text/csv",
303+
deserializer=CSVDeserializer(),
304304
)
305305

306306
# Validate that no exception is raised when the target_variant is specified.

0 commit comments

Comments
 (0)