def METHOD_NAME():
    seed = 1
    rs = np.random.RandomState(seed)
    X = rs.rand(20, 10)
    Y = rs.rand(10, 1)
    model = get_gp(10, seed)
    model.train(X[:10], Y[:10])
    m_hat, v_hat = model.predict(X[10:])
    assert m_hat.shape == (10, 1)
    assert v_hat.shape == (10, 1)