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 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):
"""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_resource_config():
test.write_json(RESOURCE_CONFIG, _env.resource_config_file_dir)
assert _env.read_resource_config() == RESOURCE_CONFIG
