Skip to content

Commit

Permalink
fix flake
Browse files Browse the repository at this point in the history
  • Loading branch information
soumith committed Dec 2, 2017
1 parent 638b10d commit 232f8c7
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions torch/distributions.py
Original file line number Diff line number Diff line change
Expand Up @@ -243,6 +243,6 @@ def sample_n(self, n):
return _standard_gamma(_expand_n(self.alpha, n)) / self.beta

def log_prob(self, value):
return (self.alpha * torch.log(self.beta)
+ (self.alpha - 1) * torch.log(value)
- self.beta * value - torch.lgamma(self.alpha))
return (self.alpha * torch.log(self.beta) +
(self.alpha - 1) * torch.log(value) -
self.beta * value - torch.lgamma(self.alpha))

0 comments on commit 232f8c7

Please sign in to comment.