a0021cb982c25923a75cd3a1eeaee00058f2cdbb,pixyz/distributions/distributions.py,DistGraph,sample,#DistGraph#Any#Any#Any#Any#Any#,132

Before Change

            input_dict = self._get_local_input_dict(values, self.graph.pred[var], var)
            option = sample_option
            option.update(self.node_option(var))
            values[var] = self._get_local_output_value(var, dist.sample(input_dict, **option))
        result_dict = delete_dict_values(values, self.marginalize_list)
        if return_all:
            output_dict = dict(delete_dict_values(x_dict, self.input_var))
            output_dict.update(result_dict)

After Change

        values = get_dict_values(x_dict, self.input_var, return_dict=True)
        for factor in self.factors(sorted=True):
            sample = self._wrapped_sample(factor, values, sample_option)
            values.update(sample)

        result_dict = delete_dict_values(values, self.marginalize_list)
        if return_all:
            output_dict = dict(delete_dict_values(x_dict, self.input_var))

Instances