Fix set union.

Joseph Zhang · Joseph Zhang · commit 09a54dc1cc10 · 2024-11-14T11:14:45.000-08:00
diff --git a/src/sagemaker/jumpstart/types.py b/src/sagemaker/jumpstart/types.py
@@ -1123,7 +1123,7 @@ class JumpStartModelDataSource(AdditionalModelDataSource):
     """Data class JumpStart additional model data source."""
 
     SERIALIZATION_EXCLUSION_SET = AdditionalModelDataSource.SERIALIZATION_EXCLUSION_SET.union(
-        "artifact_version"
+        {"artifact_version"}
     )
 
     __slots__ = list(SERIALIZATION_EXCLUSION_SET) + AdditionalModelDataSource.__slots__

Original file line number	Diff line number	Diff line change
`@@ -1123,7 +1123,7 @@ class JumpStartModelDataSource(AdditionalModelDataSource):`
`1123`	`1123`	`"""Data class JumpStart additional model data source."""`
`1124`	`1124`
`1125`	`1125`	`SERIALIZATION_EXCLUSION_SET = AdditionalModelDataSource.SERIALIZATION_EXCLUSION_SET.union(`
`1126`		`- "artifact_version"`
	`1126`	`+ {"artifact_version"}`
`1127`	`1127`	`)`
`1128`	`1128`
`1129`	`1129`	`__slots__ = list(SERIALIZATION_EXCLUSION_SET) + AdditionalModelDataSource.__slots__`