def test_list_candidates_default(sagemaker_session): auto_ml = AutoML( role=ROLE, target_attribute_name=TARGET_ATTRIBUTE_NAME, sagemaker_session=sagemaker_session ) auto_ml.current_job_name = "current_job_name" auto_ml.list_candidates() sagemaker_session.list_candidates.assert_called_once() sagemaker_session.list_candidates.assert_called_with(job_name=auto_ml.current_job_name)
def test_best_candidate_job_no_desc(sagemaker_session): auto_ml = AutoML(role=ROLE, target_attribute_name=TARGET_ATTRIBUTE_NAME, sagemaker_session=sagemaker_session) auto_ml.current_job_name = JOB_NAME best_candidate = auto_ml.best_candidate() sagemaker_session.describe_auto_ml_job.assert_called_once() sagemaker_session.describe_auto_ml_job.assert_called_with(JOB_NAME) assert best_candidate == BEST_CANDIDATE
def test_best_candidate_default_job_name(sagemaker_session): auto_ml = AutoML(role=ROLE, target_attribute_name=TARGET_ATTRIBUTE_NAME, sagemaker_session=sagemaker_session) auto_ml.current_job_name = JOB_NAME auto_ml._auto_ml_job_desc = AUTO_ML_DESC best_candidate = auto_ml.best_candidate() sagemaker_session.describe_auto_ml_job.assert_not_called() assert best_candidate == BEST_CANDIDATE