32cf78d9b668f197318b7ac7e2403506f7a167e6,mushroom_rl/solvers/lqr.py,,compute_lqr_Q,#,98

Before Change

    
    M = _compute_lqr_Q_matrix(lqr, K)
    return -s.T @ M @ s


def compute_lqg_Q(s, lqr, K, Sigma):
    

After Change

        s = s.reshape((1, -1))

    M = _compute_lqr_Q_matrix(lqr, K)
    m = lambda x, y: x.T @ M @ x
    return -1. * pairwise_distances(s, metric=m).diagonal().reshape((-1, 1))


def compute_lqg_Q(s, lqr, K, Sigma):
    

Instances