fde4b1f40f02082b56e8af3c696b3f1b0cef7fa4,src/chaospy/poly/collection/core.py,,variable,#Any#,254

Before Change

    r = np.arange(dims, dtype=int)
    A = {}
    for i in range(dims):
        A[tuple(1*(r==i))] = 1*(r==i)

    return Poly(A, dim=dims, shape=(dims,))

def order(P):

After Change

    
    if dims == 1:
        return Poly({(1,): 1}, dim=1, shape=())
    return Poly({
        tuple(indices): indices for indices in numpy.eye(dims, dtype=int)
    }, dim=dims, shape=(dims,))

def order(P):

Instances