33a33b24185e45478c4758ac14cab8ae5234c44b,pgmpy/inference/base_continuous.py,AbstractGaussian,__init__,#AbstractGaussian#Any#Any#,24

Before Change

    def __init__(self, mean_vec, cov_matrix):

        if not isinstance(mean_vec, np.matrix):
            if isinstance(mean_vec, (np.ndarray, list, tuple, set, frozenset)):
                length = len(mean_vec)
                mean_vec = np.matrix(mean_vec)
                mean_vec = np.reshape(mean_vec, (1, length))
            else:
                raise TypeError("mean_vec should be a 1d array type object")
        mean_vec = np.reshape(mean_vec, (len(mean_vec), 1))
        if not isinstance(cov_matrix, np.matrix):
            raise TypeError(
                "cov_matrix must be numpy.matrix type object")

After Change

    def __init__(self, mean_vec, cov_matrix):

        if isinstance(mean_vec, (np.matrix, np.ndarray, list, tuple, set, frozenset)):
            mean_vec = np.array(mean_vec).flatten()
        else:
            raise TypeError("mean_vec should be a 1d array type object")
        mean_vec = np.matrix(np.reshape(mean_vec, (len(mean_vec), 1)))

        if not isinstance(cov_matrix, np.matrix):
            raise TypeError(
                "cov_matrix must be numpy.matrix type object")

Instances