chore(components): SageMaker fix flaky groundtruth test (#5044)

2021-01-27 20:52:01 -08:00 · 2021-01-27 20:52:01 -08:00 · 2dd8de3f6f
parent e6aeb9cc39
commit 2dd8de3f6f
1 changed files with 1 additions and 1 deletions
--- a/components/aws/sagemaker/tests/integration_tests/component_tests/test_groundtruth_component.py
+++ b/components/aws/sagemaker/tests/integration_tests/component_tests/test_groundtruth_component.py
@ -109,7 +109,7 @@ def test_groundtruth_labeling_job(
        response = sagemaker_utils.describe_labeling_job(
            sagemaker_client, ground_truth_train_job_name
        )
-        assert response["LabelingJobStatus"] in ["Stopping", "Stopped"]
+        assert response["LabelingJobStatus"] in ["Completed", "Stopping", "Stopped"]
    finally:
        print(
            f"Clean up workteam: {workteam_arn} and GT job: {ground_truth_train_job_name}"