Skip to content

Commit a381316

Browse files
committed
temporarily skip tests impacted by data inconsistency
1 parent 242d6ed commit a381316

File tree

4 files changed

+11
-10
lines changed

4 files changed

+11
-10
lines changed

tests/integ/sagemaker/lineage/conftest.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -709,9 +709,10 @@ def static_model_artifact(sagemaker_session, static_pipeline_execution_arn):
709709
)
710710
)
711711

712-
yield artifact.ModelArtifact.load(
713-
artifacts[0]["ArtifactArn"], sagemaker_session=sagemaker_session
714-
)
712+
artifact_arn = artifacts[0]["ArtifactArn"]
713+
logging.info(f"Using static model artifact {artifact_arn}")
714+
715+
yield artifact.ModelArtifact.load(artifact_arn, sagemaker_session=sagemaker_session)
715716

716717

717718
@pytest.fixture

tests/integ/sagemaker/lineage/test_artifact.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ def test_list_by_type(artifact_objs, sagemaker_session):
101101
assert len(artifact_names_listed) == 1
102102
assert artifact_names_listed[0] == expected_name
103103

104-
104+
@pytest.mark.skip('data inconsistency P61661075')
105105
def test_get_artifact(static_dataset_artifact):
106106
s3_uri = static_dataset_artifact.source.source_uri
107107
expected_artifact = static_dataset_artifact.s3_uri_artifacts(s3_uri=s3_uri)

tests/integ/sagemaker/lineage/test_dataset_artifact.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def test_trained_models(
2626
assert model.destination_arn == model_artifact_obj1.artifact_arn
2727
assert model.destination_type == "Context"
2828

29-
29+
@pytest.mark.skip('data inconsistency P61661075')
3030
def test_endpoint_contexts(
3131
static_dataset_artifact,
3232
):
@@ -36,15 +36,15 @@ def test_endpoint_contexts(
3636
for context in contexts_from_query:
3737
assert context.context_type == "Endpoint"
3838

39-
39+
@pytest.mark.skip('data inconsistency P61661075')
4040
def test_get_upstream_datasets(static_dataset_artifact, sagemaker_session):
4141
artifacts_from_query = static_dataset_artifact.upstream_datasets()
4242
assert len(artifacts_from_query) > 0
4343
for artifact in artifacts_from_query:
4444
assert artifact.artifact_type == "DataSet"
4545
assert "artifact" in artifact.artifact_arn
4646

47-
47+
@pytest.mark.skip('data inconsistency P61661075')
4848
def test_get_down_datasets(static_dataset_artifact, sagemaker_session):
4949
artifacts_from_query = static_dataset_artifact.downstream_datasets()
5050
assert len(artifacts_from_query) > 0

tests/integ/sagemaker/lineage/test_lineage_trial_component.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -13,21 +13,21 @@
1313
"""This module contains code to test SageMaker ``Trial Component``"""
1414
from __future__ import absolute_import
1515

16-
16+
@pytest.mark.skip('data inconsistency P61661075')
1717
def test_dataset_artifacts(static_training_job_trial_component):
1818
artifacts_from_query = static_training_job_trial_component.dataset_artifacts()
1919
assert len(artifacts_from_query) > 0
2020
for artifact in artifacts_from_query:
2121
assert artifact.artifact_type == "DataSet"
2222

23-
23+
@pytest.mark.skip('data inconsistency P61661075')
2424
def test_models(static_processing_job_trial_component):
2525
artifacts_from_query = static_processing_job_trial_component.models()
2626
assert len(artifacts_from_query) > 0
2727
for artifact in artifacts_from_query:
2828
assert artifact.artifact_type == "Model"
2929

30-
30+
@pytest.mark.skip('data inconsistency P61661075')
3131
def test_pipeline_execution_arn(static_training_job_trial_component, static_pipeline_execution_arn):
3232
pipeline_execution_arn = static_training_job_trial_component.pipeline_execution_arn()
3333
assert pipeline_execution_arn == static_pipeline_execution_arn

0 commit comments

Comments
 (0)