utils.py 文件源码

def __call__(self, *args, **kwargs):
    underlying_frozen = self.underlying(*args, **kwargs)
    pmf = binarize_pdf(underlying_frozen, limit=self.max_value + 1)

    return stats.rv_discrete(
      a=0,
      b=self.max_value,
      values=(np.arange(self.max_value + 1), pmf)
    )