e1f44ea0337ace78ef8b796439139acbec60dd37,pyro/contrib/forecast/forecaster.py,HMCForecaster,forward,#HMCForecaster#Any#Any#Any#Any#,432

Before Change

                node["value"] = sample.reshape(
                    (num_samples,) + (1,) * (node["value"].dim() - sample.dim()) + sample.shape[1:])

            with PrefixReplayMessenger(self._trace):
                with PrefixConditionMessenger(self.model._prefix_condition_data):
                    with pyro.plate("particles", num_samples, dim=dim):
                        return self.model(data, covariates)

After Change

                node["value"] = sample.reshape(
                    (num_samples,) + (1,) * (node["value"].dim() - sample.dim()) + sample.shape[1:])

            with ExitStack() as stack:
                if data.size(-2) < covariates.size(-2):
                    stack.enter_context(PrefixReplayMessenger(self._trace))
                    stack.enter_context(
                        PrefixConditionMessenger(self.model._prefix_condition_data))
                with pyro.plate("particles", num_samples, dim=dim):
                    return self.model(data, covariates)

Instances