def test_warm_start_config_init(type, parents):
warm_start_config = WarmStartConfig(warm_start_type=type, parents=parents)
assert warm_start_config.type == type, "Warm start type initialization failed."
assert warm_start_config.parents == set(parents), "Warm start parents config initialization failed."
warm_start_config_req = warm_start_config.to_input_req()
assert warm_start_config.type == WarmStartTypes(warm_start_config_req["WarmStartType"])
for parent in warm_start_config_req["ParentHyperParameterTuningJobs"]:
assert parent['HyperParameterTuningJobName'] in parents
def test_attach_with_warm_start_config(sagemaker_session):
warm_start_config = WarmStartConfig(warm_start_type=WarmStartTypes.TRANSFER_LEARNING, parents={"p1", "p2"})
job_details = copy.deepcopy(TUNING_JOB_DETAILS)
job_details["WarmStartConfig"] = warm_start_config.to_input_req()
sagemaker_session.sagemaker_client.describe_hyper_parameter_tuning_job = Mock(name='describe_tuning_job',
return_value=job_details)
tuner = HyperparameterTuner.attach(JOB_NAME, sagemaker_session=sagemaker_session)
assert tuner.warm_start_config.type == warm_start_config.type
assert tuner.warm_start_config.parents == warm_start_config.parents
