Python Predictor.regress Examples

Example #1

File: test_tfs.py Project: xuefeng7/sagemaker-python-sdk

def test_predictor_regress(sagemaker_session):
    predictor = Predictor('endpoint', sagemaker_session, model_name='model', model_version='123')

    mock_response(json.dumps(REGRESS_RESPONSE).encode('utf-8'), sagemaker_session)
    result = predictor.regress(REGRESS_INPUT)

    assert_invoked_with_body_dict(sagemaker_session,
                                  EndpointName='endpoint',
                                  ContentType=JSON_CONTENT_TYPE,
                                  Accept=JSON_CONTENT_TYPE,
                                  CustomAttributes='tfs-method=regress,tfs-model-name=model,tfs-model-version=123',
                                  Body=json.dumps(REGRESS_INPUT))

    assert REGRESS_RESPONSE == result

Example #2

def test_predictor_regress(sagemaker_session):
    predictor = Predictor("endpoint", sagemaker_session, model_name="model", model_version="123")

    mock_response(json.dumps(REGRESS_RESPONSE).encode("utf-8"), sagemaker_session)
    result = predictor.regress(REGRESS_INPUT)

    assert_invoked_with_body_dict(
        sagemaker_session,
        EndpointName="endpoint",
        ContentType=JSON_CONTENT_TYPE,
        Accept=JSON_CONTENT_TYPE,
        CustomAttributes="tfs-method=regress,tfs-model-name=model,tfs-model-version=123",
        Body=json.dumps(REGRESS_INPUT),
    )

    assert REGRESS_RESPONSE == result

Example #3

File: test_tfs.py Project: wqzoww/sagemaker-python-sdk

def test_predictor_regress_bad_content_type(sagemaker_session):
    predictor = Predictor("endpoint", sagemaker_session, csv_serializer)

    with pytest.raises(ValueError):
        predictor.regress(REGRESS_INPUT)