def augment_data(self, augmented_data):
        """
        Augment the data with auxiliary variables
        :param augmented_data:
        :return:
        """
        x = augmented_data["x"]
        T, K = x.shape
        assert K <= self.K

        augmented_data["kappa"] = kappa_vec(x)
        augmented_data["omega"] = np.ones((T,K-1))

        self.resample_omega([augmented_data])

        return augmented_data
dataset/ETHPy150Open HIPS/pgmult/pgmult/distributions.py/PGMultinomialRegression.augment_data/operator `==` -> `<=`