def test_candidate_estimator_get_steps(sagemaker_session): auto_ml = AutoML(role=ROLE, target_attribute_name=TARGET_ATTRIBUTE_NAME, sagemaker_session=sagemaker_session) candidates = auto_ml.list_candidates(job_name=AUTO_ML_JOB_NAME) candidate = candidates[1] candidate_estimator = CandidateEstimator(candidate, sagemaker_session) steps = candidate_estimator.get_steps() assert len(steps) == 3
def test_candidate_estimator_get_steps(sagemaker_session): candidate_estimator = CandidateEstimator( CANDIDATE_DICT, sagemaker_session=sagemaker_session) steps = candidate_estimator.get_steps() assert len(steps) == 3