|
34 | 34 | ("sagemaker.predictor._NumpyDeserializer()", True),
|
35 | 35 | ("sagemaker.predictor._JsonDeserializer()", True),
|
36 | 36 | ("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), |
39 | 39 | ("_CsvSerializer()", True),
|
40 | 40 | ("_JsonSerializer()", True),
|
41 | 41 | ("_NpySerializer()", True),
|
@@ -250,8 +250,8 @@ def test_import_from_predictor_modify_node(src, expected):
|
250 | 250 | @pytest.mark.parametrize(
|
251 | 251 | "import_statement, expected",
|
252 | 252 | [
|
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), |
255 | 255 | ("from sagemaker.amazon.common import write_spmatrix_to_sparse_tensor", False),
|
256 | 256 | ],
|
257 | 257 | )
|
|
0 commit comments