Improvements of the integration tests

Author: Anton Repushko

tests/integ/auto_ml_v2_utils.py

@@ -69,6 +69,7 @@
         item_identifier_attribute_name="item_id",
         target_attribute_name="target",
         timestamp_attribute_name="timestamp",
+        forecast_quantiles=["p10", "p50", "p90"],
     ),
 }
 

tests/integ/test_auto_ml_v2.py

@@ -251,6 +251,8 @@ def test_list_candidates(
 
         candidates = auto_ml.list_candidates(job_name=job_name)
         assert len(candidates) == num_candidates
+    else:
+        pytest.skip("The job hasn't finished yet")
 
 
 @pytest.mark.skipif(
@@ -320,6 +322,8 @@ def test_best_candidate(
         best_candidate = auto_ml.best_candidate(job_name=job_name)
         assert len(best_candidate["InferenceContainers"]) == num_containers
         assert best_candidate["CandidateStatus"] == "Completed"
+    else:
+        pytest.skip("The job hasn't finished yet")
 
 
 @pytest.mark.skipif(
@@ -411,6 +415,8 @@ def test_deploy_best_candidate(
         )["EndpointStatus"]
         assert endpoint_status == "InService"
         sagemaker_session.sagemaker_client.delete_endpoint(EndpointName=endpoint_name)
+    else:
+        pytest.skip("The job hasn't finished yet")
 
 
 @pytest.mark.skipif(
@@ -482,3 +488,5 @@ def test_candidate_estimator_get_steps(
         candidate_estimator = CandidateEstimator(candidate, sagemaker_session)
         steps = candidate_estimator.get_steps()
         assert len(steps) == num_steps
+    else:
+        pytest.skip("The job hasn't finished yet")