Python AutoMLJob.start_new Beispiele

Beispiel #1

Datei: test_auto_ml.py Projekt: loicdiridollou/sagemaker-python-sdk

def test_auto_ml_invalid_input_data_format(sagemaker_session):
    auto_ml = AutoML(role=ROLE,
                     target_attribute_name=TARGET_ATTRIBUTE_NAME,
                     sagemaker_session=sagemaker_session)
    inputs = {}

    expected_error_msg = "Cannot format input {}. Expecting one of str or list of strings."
    with pytest.raises(ValueError, message=expected_error_msg.format(inputs)):
        AutoMLJob.start_new(auto_ml, inputs)
    sagemaker_session.auto_ml.assert_not_called()

Beispiel #2

Datei: test_auto_ml.py Projekt: rohangujarathi/sagemaker-python-sdk

def test_auto_ml_default_channel_name(sagemaker_session):
    auto_ml = AutoML(
        role=ROLE, target_attribute_name=TARGET_ATTRIBUTE_NAME, sagemaker_session=sagemaker_session
    )
    inputs = DEFAULT_S3_INPUT_DATA
    AutoMLJob.start_new(auto_ml, inputs)
    sagemaker_session.auto_ml.assert_called_once()
    _, args = sagemaker_session.auto_ml.call_args
    assert args["input_config"] == [
        {
            "DataSource": {
                "S3DataSource": {"S3DataType": "S3Prefix", "S3Uri": DEFAULT_S3_INPUT_DATA}
            },
            "TargetAttributeName": TARGET_ATTRIBUTE_NAME,
        }
    ]