e46c499eb6f5a8bf8c1abd914e9e5b6818f07d1a,pyprob/state.py,,sample,#Any#Any#Any#Any#,82

Before Change

            current_sample = Sample(distribution, distribution.sample(), address, instance, log_prob=0, control=control, replace=replace)
        else:  // _trace_mode == TraceMode.LIGHTWEIGHT_METROPOLIS_HASTINGS:
            current_sample = Sample(distribution, distribution.sample(), address, instance, control=True, replace=False)
            if _metropolis_hastings_trace is not None:
                if current_sample.address == _metropolis_hastings_proposal_address or current_sample.address not in _metropolis_hastings_trace._samples_all_dict_address:
                    value = distribution.sample()
                    reused = False
                else:
                    value = _metropolis_hastings_trace._samples_all_dict_address[current_sample.address].value
                    reused = True
                current_sample = Sample(distribution, value, address, instance, control=True, replace=False, reused=reused)

        _current_trace.add_sample(current_sample)
        return current_sample.value

After Change

                    value = _metropolis_hastings_trace._samples_all_dict_address[current_sample.address].value
                    reused = True
                    try:  // Takes care of issues such as changed distribution parameters (e.g., batch size) that prevent a rescoring of a reused value under this distribution.
                        log_prob = distribution.log_prob(value)
                    except:
                        value = distribution.sample()
                        log_prob = distribution.log_prob(value)
                        reused = False

Instances