Skip to content

Commit 9ec131e

Browse files
committed
Fix repr
1 parent f42ba6c commit 9ec131e

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

src/sagemaker/serve/builder/schema_builder.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -208,13 +208,14 @@ def _get_inverse(self, obj):
208208

209209
def __repr__(self):
210210
"""Placeholder docstring"""
211-
return (
212-
f"SchemaBuilder(\n"
213-
f"input_serializer={self.input_serializer}\n"
214-
f"output_serializer={self.output_serializer}\n"
215-
f"input_deserializer={self.input_deserializer._deserializer}\n"
216-
f"output_deserializer={self.output_deserializer._deserializer})"
217-
)
211+
if hasattr(self, "input_serializer") and hasattr(self, "output_serializer"):
212+
return (
213+
f"SchemaBuilder(\n"
214+
f"input_serializer={self.input_serializer}\n"
215+
f"output_serializer={self.output_serializer}\n"
216+
f"input_deserializer={self.input_deserializer._deserializer}\n"
217+
f"output_deserializer={self.output_deserializer._deserializer})"
218+
)
218219

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

0 commit comments

Comments
 (0)