35110fa3d224a0d72638f3011a08640e237c2416,pyro/contrib/epidemiology/compartmental.py,CompartmentalModel,fit,#CompartmentalModel#,283

Before Change

        // Save these options for .predict().
        self.num_quant_bins = options.pop("num_quant_bins", 4)
        self._dct = options.pop("dct", None)
        if self._dct is not None and self.is_regional:
            raise NotImplementedError("regional models do not support DiscreteCosineReparam")

        // Heuristically initialze to feasible latents.
        heuristic_options = {k.replace("heuristic_", ""): options.pop(k)
                             for k in list(options)
                             if k.startswith("heuristic_")}

After Change

        
        // Save these options for .predict().
        self.num_quant_bins = options.pop("num_quant_bins", 4)
        haar = options.pop("haar", False)

        // Heuristically initialze to feasible latents.
        heuristic_options = {k.replace("heuristic_", ""): options.pop(k)
                             for k in list(options)

Instances