f7a56772232dcf8328261f69538e0fee9bb88d3c,pyro/contrib/epidemiology/compartmental.py,CompartmentalModel,_quantized_model,#CompartmentalModel#,680

Before Change

        // Sample any non-compartmental time series in batch.
        // TODO Consider using pyro.contrib.forecast.util.reshape_batch to
        // support DiscreteCosineReparam and HaarReparam along the time dim.
        for name, (fn, is_regional) in self._non_compartmental.items():
            fn = dist.ImproperUniform(fn.support, fn.batch_shape, fn.event_shape)
            with self.time_plate, optional(self.region_plate, is_regional):
                curr[name] = pyro.sample(name, fn)

        // Truncate final value from the right then pad initial value onto the left.
        init = self.initialize(params)
        prev = {}
        for name, value in init.items():
            if name in self.compartments:

After Change

        // Sample global parameters and auxiliary variables.
        params = self.global_model()
        auxiliary, non_compartmental = self._sample_auxiliary()

        // Manually enumerate.
        curr, logp = quantize_enumerate(auxiliary, min=0, max=self.population,

Instances