def test_prepare_warm_start_config_cls(warm_start_config_req):
warm_start_config = WarmStartConfig.from_job_desc(warm_start_config_req)
assert warm_start_config.type == WarmStartTypes(
warm_start_config_req["WarmStartType"]
), "Warm start type initialization failed."
for p in warm_start_config_req["ParentHyperParameterTuningJobs"]:
assert (
p["HyperParameterTuningJobName"] in warm_start_config.parents
), "Warm start parents config initialization failed."
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_tune_warm_start(sagemaker_session, warm_start_type, parents):
def assert_create_tuning_job_request(**kwrags):
assert (kwrags["HyperParameterTuningJobConfig"] ==
SAMPLE_TUNING_JOB_REQUEST["HyperParameterTuningJobConfig"])
assert kwrags["HyperParameterTuningJobName"] == "dummy-tuning-1"
assert kwrags["TrainingJobDefinition"] == SAMPLE_TUNING_JOB_REQUEST[
"TrainingJobDefinition"]
assert kwrags["WarmStartConfig"] == {
"WarmStartType":
warm_start_type,
"ParentHyperParameterTuningJobs": [{
"HyperParameterTuningJobName":
parent
} for parent in parents],
}
sagemaker_session.sagemaker_client.create_hyper_parameter_tuning_job.side_effect = (
assert_create_tuning_job_request)
sagemaker_session.tune(
job_name="dummy-tuning-1",
strategy="Bayesian",
objective_type="Maximize",
objective_metric_name="val-score",
max_jobs=100,
max_parallel_jobs=5,
parameter_ranges=SAMPLE_PARAM_RANGES,
static_hyperparameters=STATIC_HPs,
image="dummy-image-1",
input_mode="File",
metric_definitions=SAMPLE_METRIC_DEF,
role=EXPANDED_ROLE,
input_config=SAMPLE_INPUT,
output_config=SAMPLE_OUTPUT,
resource_config=RESOURCE_CONFIG,
stop_condition=SAMPLE_STOPPING_CONDITION,
tags=None,
warm_start_config=WarmStartConfig(
warm_start_type=WarmStartTypes(warm_start_type),
parents=parents).to_input_req(),
)