def test_description():
session = create_sagemaker_session(describe_tuning_result={
'HyperParameterTuningJobConfig': {
'ParameterRanges': {
'CategoricalParameterRanges': [],
'ContinuousParameterRanges': [
{'MaxValue': '1', 'MinValue': '0', 'Name': 'eta'},
{'MaxValue': '10', 'MinValue': '0', 'Name': 'gamma'},
],
'IntegerParameterRanges': [
{'MaxValue': '30', 'MinValue': '5', 'Name': 'num_layers'},
{'MaxValue': '100', 'MinValue': '50', 'Name': 'iterations'},
],
},
},
})
tuner = HyperparameterTuningJobAnalytics("my-tuning-job", sagemaker_session=session)
d = tuner.description()
assert len(session.sagemaker_client.describe_hyper_parameter_tuning_job.mock_calls) == 1
assert d is not None
assert d['HyperParameterTuningJobConfig'] is not None
tuner.clear_cache()
d = tuner.description()
assert len(session.sagemaker_client.describe_hyper_parameter_tuning_job.mock_calls) == 2
d = tuner.description()
assert len(session.sagemaker_client.describe_hyper_parameter_tuning_job.mock_calls) == 2
d = tuner.description(force_refresh=True)
assert len(session.sagemaker_client.describe_hyper_parameter_tuning_job.mock_calls) == 3
# Check that the ranges work.
r = tuner.tuning_ranges
assert len(r) == 4
def test_description():
session = create_sagemaker_session(
describe_tuning_result={
"HyperParameterTuningJobConfig": {
"ParameterRanges": {
"CategoricalParameterRanges": [],
"ContinuousParameterRanges": [
{
"MaxValue": "1",
"MinValue": "0",
"Name": "eta"
},
{
"MaxValue": "10",
"MinValue": "0",
"Name": "gamma"
},
],
"IntegerParameterRanges": [
{
"MaxValue": "30",
"MinValue": "5",
"Name": "num_layers"
},
{
"MaxValue": "100",
"MinValue": "50",
"Name": "iterations"
},
],
}
}
})
tuner = HyperparameterTuningJobAnalytics("my-tuning-job",
sagemaker_session=session)
d = tuner.description()
assert len(session.sagemaker_client.describe_hyper_parameter_tuning_job.
mock_calls) == 1
assert d is not None
assert d["HyperParameterTuningJobConfig"] is not None
tuner.clear_cache()
d = tuner.description()
assert len(session.sagemaker_client.describe_hyper_parameter_tuning_job.
mock_calls) == 2
d = tuner.description()
assert len(session.sagemaker_client.describe_hyper_parameter_tuning_job.
mock_calls) == 2
d = tuner.description(force_refresh=True)
assert len(session.sagemaker_client.describe_hyper_parameter_tuning_job.
mock_calls) == 3
# Check that the ranges work.
r = tuner.tuning_ranges
assert len(r) == 4
