Skip to content

Commit 8cc39f0

Browse files
committed
more logging when unexpected number of artifacts found
1 parent d431778 commit 8cc39f0

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/integ/sagemaker/lineage/conftest.py

+8
Original file line numberDiff line numberDiff line change
@@ -664,6 +664,9 @@ def static_endpoint_context(sagemaker_session, static_pipeline_execution_arn):
664664
"ContextSummaries"
665665
]
666666
if len(contexts) != 1:
667+
for ctx in context:
668+
logging.info(f'Found context "{ctx["ContextArn"]}"')
669+
667670
raise (
668671
Exception(
669672
f"Got an unexpected number of Contexts for \
@@ -713,7 +716,12 @@ def static_model_artifact(sagemaker_session, static_pipeline_execution_arn):
713716
artifacts = sagemaker_session.sagemaker_client.list_artifacts(SourceUri=model_package_arn)[
714717
"ArtifactSummaries"
715718
]
719+
716720
if len(artifacts) != 1:
721+
722+
for art in artifacts:
723+
logging.info(f'Found artifact {art["ArtifactArn"]}')
724+
717725
raise (
718726
Exception(
719727
f"Got an unexpected number of Artifacts for \

0 commit comments

Comments
 (0)