Skip to content

Commit 73bc6c4

Browse files
committed
Fix repr
1 parent 9ec131e commit 73bc6c4

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/sagemaker/serve/builder/schema_builder.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -216,6 +216,12 @@ def __repr__(self):
216216
f"input_deserializer={self.input_deserializer._deserializer}\n"
217217
f"output_deserializer={self.output_deserializer._deserializer})"
218218
)
219+
elif hasattr(self, "custom_input_translator") and hasattr(self, "custom_output_translator"):
220+
return (
221+
f"SchemaBuilder(\n"
222+
f"custom_input_translator={self.custom_input_translator}\n"
223+
f"custom_output_translator={self.custom_output_translator}\n"
224+
)
219225

220226
def generate_marshalling_map(self) -> dict:
221227
"""Generate marshalling map for the schema builder"""

0 commit comments

Comments
 (0)