a0021cb982c25923a75cd3a1eeaee00058f2cdbb,pixyz/distributions/distributions.py,DistGraph,var_replaced,#DistGraph#Any#Any#,75

Before Change

                name_dict[new_global_var] = avar

    def var_replaced(self, replace_dict, name=""):
        if not (set(replace_dict.keys()) <= set(self.graph)):
            unknown_var = [var_name for var_name in replace_dict.keys() if var_name not in self.graph]
            raise ValueError(f"replace_dict has unknown variables: {unknown_var}")
        if not set(replace_dict.values()).isdisjoint(set(self.graph)):

After Change

        result.marginalize_list = {replace_dict[var] if var in replace_dict else var for var in self.marginalize_list}
        result.global_option = dict(self.global_option)

        for factor in result.factors():
            if set(replace_dict.values()).isdisjoint(list(result.graph.pred[factor]) + list(result.graph.succ[factor])):
                continue
            factor.rename_var(replace_dict)
        return result

    def _factors_from_variable(self, var_name):
        return list(self.graph.pred[var_name])

Instances