Python default_input_fn Examples

Example #1

File: test_serving.py Project: rizwangilani/sagemaker-xgboost-container

def test_input_fn_bad_content_type():
    with pytest.raises(errors.UnsupportedFormatError):
        serving.default_input_fn('', 'application/not_supported')

Example #2

File: test_serving.py Project: rizwangilani/sagemaker-xgboost-container

def test_input_fn_dmatrix(csv_array):
    deserialized_csv_array = serving.default_input_fn(csv_array,
                                                      content_types.CSV)
    assert type(deserialized_csv_array) is xgb.DMatrix