33573192e348ad5e1dfe188df6f3fa2a7a892372,pixyz/distributions/distributions.py,DistributionBase,get_params,#DistributionBase#,774

Before Change

    @lru_cache_for_sample_dict()
    def get_params(self, params_dict={}, **kwargs):
        params_dict, vars_dict = replace_dict_keys_split(params_dict, self.replace_params_dict)
        output_dict = self.forward(**vars_dict)

        output_dict.update(params_dict)

After Change

    @lru_cache_for_sample_dict()
    def get_params(self, params_dict={}, **kwargs):
        replaced_params_dict = {}
        for key, value in params_dict.items():
            if key in self.replace_params_dict:
                for replaced_key in self.replace_params_dict[key]:
                    replaced_params_dict[replaced_key] = value

        vars_dict = {key: value for key, value in params_dict.items() if key not in self.replace_params_dict}

        output_dict = self.forward(**vars_dict)

Instances