4d77268c95b8a041a3bac848860765afdad1af46,pymc3/distributions/discrete.py,DiscreteUniform,__init__,#DiscreteUniform#Any#Any#,300

Before Change

    
    def __init__(self, lower, upper, *args, **kwargs):
        super(DiscreteUniform, self).__init__(*args, **kwargs)
        self.lower = T.floor(lower).astype("int32")
        self.upper = T.floor(upper).astype("int32")
        self.mode = T.floor((upper - lower) / 2.).astype("int32")

    def _random(self, lower, upper, size=None):
        // This way seems to be the only to deal with lower and upper

After Change

    
    def __init__(self, lower, upper, *args, **kwargs):
        super(DiscreteUniform, self).__init__(*args, **kwargs)
        self.lower, self.upper = floor(lower).astype("int32"), floor(upper).astype("int32")
        self.mode = floor((upper - lower) / 2.).astype("int32")

    def _random(self, lower, upper, size=None):
        // This way seems to be the only to deal with lower and upper as array-like.

Instances