@pytest.mark.skipif((tests.integ.test_region() in tests.integ.NO_AUTO_ML_REGIONS), reason='AutoML is not supported in the region yet.')
def test_best_candidate(sagemaker_session):
    auto_ml_utils.create_auto_ml_job_if_not_exist(sagemaker_session)
    auto_ml = AutoML(role=ROLE, target_attribute_name=TARGET_ATTRIBUTE_NAME, sagemaker_session=sagemaker_session)
    best_candidate = auto_ml.best_candidate(job_name=AUTO_ML_JOB_NAME)
    assert (len(best_candidate['InferenceContainers']) == 3)
    assert (len(best_candidate['CandidateSteps']) == 4)
    assert (best_candidate['CandidateStatus'] == 'Completed')
@pytest.mark.skipif((tests.integ.test_region() in tests.integ.NO_AUTO_ML_REGIONS), reason='AutoML is not supported in the region yet.')
def test_best_candidate(sagemaker_session):
    auto_ml_utils.create_auto_ml_job_if_not_exist(sagemaker_session)
    auto_ml = AutoML(role=ROLE, target_attribute_name=TARGET_ATTRIBUTE_NAME, sagemaker_session=sagemaker_session)
    best_candidate = auto_ml.best_candidate(job_name=AUTO_ML_JOB_NAME)
    assert (len(best_candidate['InferenceContainers']) == 3)
    assert (len(best_candidate['CandidateSteps']) == 4)
    assert (best_candidate['CandidateStatus'] == 'Completed')