Skip to content

Commit 64a91ce

Browse files
authored
fix: update source constructor for lineage action and artifact (#2095)
1 parent 49cab87 commit 64a91ce

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/sagemaker/lineage/action.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ def create(
186186
return super(Action, cls)._construct(
187187
cls._boto_create_method,
188188
action_name=action_name,
189-
source=_api_types.ContextSource(source_uri=source_uri, source_type=source_type),
189+
source=_api_types.ActionSource(source_uri=source_uri, source_type=source_type),
190190
action_type=action_type,
191191
description=description,
192192
status=status,

src/sagemaker/lineage/artifact.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -258,7 +258,7 @@ def create(
258258
return super(Artifact, cls)._construct(
259259
cls._boto_create_method,
260260
artifact_name=artifact_name,
261-
source=_api_types.ContextSource(source_uri=source_uri, source_types=source_types),
261+
source=_api_types.ArtifactSource(source_uri=source_uri, source_types=source_types),
262262
artifact_type=artifact_type,
263263
properties=properties,
264264
tags=tags,

0 commit comments

Comments
 (0)