Skip to content

Commit a33b80d

Browse files
committed
chore: expose raw spec json
1 parent 4c5dd1f commit a33b80d

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/sagemaker/jumpstart/types.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1258,9 +1258,10 @@ class JumpStartMetadataBaseFields(JumpStartDataHolderType):
12581258
"hosting_neuron_model_version",
12591259
"hub_content_type",
12601260
"_is_hub_content",
1261+
"_raw_json",
12611262
]
12621263

1263-
_non_serializable_slots = ["_is_hub_content"]
1264+
_non_serializable_slots = ["_is_hub_content", "_raw_json"]
12641265

12651266
def __init__(self, fields: Dict[str, Any], is_hub_content: Optional[bool] = False):
12661267
"""Initializes a JumpStartMetadataFields object.
@@ -1269,6 +1270,7 @@ def __init__(self, fields: Dict[str, Any], is_hub_content: Optional[bool] = Fals
12691270
fields (Dict[str, Any]): Dictionary representation of metadata fields.
12701271
"""
12711272
self._is_hub_content = is_hub_content
1273+
self._raw_json = fields
12721274
self.from_json(fields)
12731275

12741276
def from_json(self, json_obj: Dict[str, Any]) -> None:

0 commit comments

Comments
 (0)