def training_env():
"""Create a TrainingEnv.
Returns:
TrainingEnv: an instance of TrainingEnv
"""
from sagemaker_containers import _env
return _env.TrainingEnv(resource_config=_env.read_resource_config(),
input_data_config=_env.read_input_data_config(),
hyperparameters=_env.read_hyperparameters())
def training_env(resource_config=None,
input_data_config=None,
hyperparameters=None):
resource_config = resource_config or env.read_resource_config()
input_data_config = input_data_config or env.read_input_data_config()
hyperparameters = hyperparameters or env.read_hyperparameters()
return env.TrainingEnv(resource_config=resource_config,
input_data_config=input_data_config,
hyperparameters=hyperparameters)
def test_read_value_serialized_and_non_value_serialized_hyperparameters():
hyperparameters = {
k: json.dumps(v)
for k, v in SAGEMAKER_HYPERPARAMETERS.items()
}
hyperparameters.update(USER_HYPERPARAMETERS)
test.write_json(hyperparameters, _env.hyperparameters_file_dir)
assert _env.read_hyperparameters() == ALL_HYPERPARAMETERS
def training_env(resource_config=None, input_data_config=None, hyperparameters=None):
"""Placeholder docstring"""
resource_config = resource_config or env.read_resource_config()
input_data_config = input_data_config or env.read_input_data_config()
hyperparameters = hyperparameters or env.read_hyperparameters()
return env.TrainingEnv(
resource_config=resource_config,
input_data_config=input_data_config,
hyperparameters=hyperparameters,
)
def test_read_exception(loads):
loads.side_effect = ValueError('Unable to read.')
assert _env.read_hyperparameters() == {'a': 1}
def test_read_value_serialized_hyperparameters():
serialized_hps = {k: json.dumps(v) for k, v in ALL_HYPERPARAMETERS.items()}
test.write_json(serialized_hps, _env.hyperparameters_file_dir)
assert _env.read_hyperparameters() == ALL_HYPERPARAMETERS
def test_read_hyperparameters():
test.write_json(ALL_HYPERPARAMETERS, _env.hyperparameters_file_dir)
assert _env.read_hyperparameters() == ALL_HYPERPARAMETERS
