temporarily skip tests impacted by data inconsistency

Author: danabens

tests/integ/sagemaker/lineage/conftest.py

@@ -709,9 +709,10 @@ def static_model_artifact(sagemaker_session, static_pipeline_execution_arn):
             )
         )
 
-    yield artifact.ModelArtifact.load(
-        artifacts[0]["ArtifactArn"], sagemaker_session=sagemaker_session
-    )
+    artifact_arn = artifacts[0]["ArtifactArn"]
+    logging.info(f"Using static model artifact {artifact_arn}")
+
+    yield artifact.ModelArtifact.load(artifact_arn, sagemaker_session=sagemaker_session)
 
 
 @pytest.fixture

tests/integ/sagemaker/lineage/test_artifact.py

@@ -101,7 +101,7 @@ def test_list_by_type(artifact_objs, sagemaker_session):
     assert len(artifact_names_listed) == 1
     assert artifact_names_listed[0] == expected_name
 
-
+@pytest.mark.skip('data inconsistency P61661075')
 def test_get_artifact(static_dataset_artifact):
     s3_uri = static_dataset_artifact.source.source_uri
     expected_artifact = static_dataset_artifact.s3_uri_artifacts(s3_uri=s3_uri)

tests/integ/sagemaker/lineage/test_dataset_artifact.py

@@ -13,6 +13,7 @@
 """This module contains code to test SageMaker ``DatasetArtifact``"""
 from __future__ import absolute_import
 
+import pytest
 
 def test_trained_models(
     dataset_artifact_associated_models,
@@ -26,7 +27,7 @@ def test_trained_models(
         assert model.destination_arn == model_artifact_obj1.artifact_arn
         assert model.destination_type == "Context"
 
-
+@pytest.mark.skip('data inconsistency P61661075')
 def test_endpoint_contexts(
     static_dataset_artifact,
 ):
@@ -36,15 +37,15 @@ def test_endpoint_contexts(
     for context in contexts_from_query:
         assert context.context_type == "Endpoint"
 
-
+@pytest.mark.skip('data inconsistency P61661075')
 def test_get_upstream_datasets(static_dataset_artifact, sagemaker_session):
     artifacts_from_query = static_dataset_artifact.upstream_datasets()
     assert len(artifacts_from_query) > 0
     for artifact in artifacts_from_query:
         assert artifact.artifact_type == "DataSet"
         assert "artifact" in artifact.artifact_arn
 
-
+@pytest.mark.skip('data inconsistency P61661075')
 def test_get_down_datasets(static_dataset_artifact, sagemaker_session):
     artifacts_from_query = static_dataset_artifact.downstream_datasets()
     assert len(artifacts_from_query) > 0

tests/integ/sagemaker/lineage/test_lineage_trial_component.py

@@ -13,21 +13,23 @@
 """This module contains code to test SageMaker ``Trial Component``"""
 from __future__ import absolute_import
 
+import pytest
 
+@pytest.mark.skip('data inconsistency P61661075')
 def test_dataset_artifacts(static_training_job_trial_component):
     artifacts_from_query = static_training_job_trial_component.dataset_artifacts()
     assert len(artifacts_from_query) > 0
     for artifact in artifacts_from_query:
         assert artifact.artifact_type == "DataSet"
 
-
+@pytest.mark.skip('data inconsistency P61661075')
 def test_models(static_processing_job_trial_component):
     artifacts_from_query = static_processing_job_trial_component.models()
     assert len(artifacts_from_query) > 0
     for artifact in artifacts_from_query:
         assert artifact.artifact_type == "Model"
 
-
+@pytest.mark.skip('data inconsistency P61661075')
 def test_pipeline_execution_arn(static_training_job_trial_component, static_pipeline_execution_arn):
     pipeline_execution_arn = static_training_job_trial_component.pipeline_execution_arn()
     assert pipeline_execution_arn == static_pipeline_execution_arn