Exemplo n.º 1
0
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)
Exemplo n.º 2
0
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())
Exemplo n.º 3
0
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,
    )
Exemplo n.º 4
0
def test_input_data_config():
    test.write_json(INPUT_DATA_CONFIG, _env.input_data_config_file_dir)

    assert _env.read_input_data_config() == INPUT_DATA_CONFIG