Skip to content

Commit e7a22cd

Browse files
author
pintaoz
committed
fix test
1 parent 569c35e commit e7a22cd

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/unit/sagemaker/cli/compatibility/v2/modifiers/test_serde.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,8 @@
3434
("sagemaker.predictor._NumpyDeserializer()", True),
3535
("sagemaker.predictor._JsonDeserializer()", True),
3636
("sagemaker.predictor.OtherClass()", False),
37-
("sagemaker.serializers.numpy_to_record_serializer()", True),
38-
("sagemaker.deserializers.record_deserializer()", True),
37+
("sagemaker.amazon.common.numpy_to_record_serializer()", True),
38+
("sagemaker.amazon.common.record_deserializer()", True),
3939
("_CsvSerializer()", True),
4040
("_JsonSerializer()", True),
4141
("_NpySerializer()", True),
@@ -250,8 +250,8 @@ def test_import_from_predictor_modify_node(src, expected):
250250
@pytest.mark.parametrize(
251251
"import_statement, expected",
252252
[
253-
("from sagemaker.serializers import numpy_to_record_serializer", True),
254-
("from sagemaker.deserializers import record_deserializer", True),
253+
("from sagemaker.amazon.common import numpy_to_record_serializer", True),
254+
("from sagemaker.amazon.common import record_deserializer", True),
255255
("from sagemaker.amazon.common import write_spmatrix_to_sparse_tensor", False),
256256
],
257257
)

0 commit comments

Comments
 (0)