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