from dl.src.info.base import BaseInfo


class CLGVAEInfo(BaseInfo):
    def __init__(self, args, **kwargs):
        super(CLGVAEInfo, self).__init__(args, **kwargs)
        self.hy_hes = args.hy_hes
        self.hy_rec = args.hy_rec
        self.hy_commute = args.hy_commute
        self.forward_eq_prob = args.forward_eq_prob
        self.subgroup_sizes_ls = args.subgroup_sizes_ls
        self.subspace_sizes_ls = args.subspace_sizes_ls

        self.elbo = kwargs["elbo"]
        self.obj = kwargs['obj']
        self.reconst = kwargs["reconst"]
        self.kld = kwargs["kld"]
        self.group = kwargs["group"]