Skip to content

Commit c966560

Browse files
committed
Adding default identity implementations to InferenceSpec
1 parent bcd5348 commit c966560

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/sagemaker/serve/spec/inference_spec.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,9 +30,11 @@ def invoke(self, input_object: object, model: object):
3030

3131
def preprocess(self, input_data: object):
3232
"""Custom pre-processing function"""
33+
return input_data
3334

3435
def postprocess(self, predictions: object):
3536
"""Custom post-processing function"""
37+
return predictions
3638

3739
def prepare(self, *args, **kwargs):
3840
"""Custom prepare function"""

0 commit comments

Comments
 (0)