044fb774095cdf8b4166204878eb338127f18871,polylearn/tests/test_polynomial_network.py,,test_convergence_warning,#,121

Before Change

def test_convergence_warning():
    degree = 4
    y = np.product(np.dot(U[:degree], X.T), axis=0).sum(axis=0)

    est = PolynomialNetworkRegressor(degree=degree, n_components=n_components,
                                     beta=1e-10, max_iter=1, tol=1e-5,
                                     random_state=0)

After Change

def test_convergence_warning():
    degree = 4
    y = _lifted_predict(U[:degree], X)

    est = PolynomialNetworkRegressor(degree=degree, n_components=n_components,
                                     beta=1e-10, max_iter=1, tol=1e-5,
                                     random_state=0)

Instances