Can't reproduce TweedieDevianceScore for power=1 using formula stated in docs

[EvanZ]

Here's an example with output:

deviance = TweedieDevianceScore(power=1)
yhat = torch.tensor([0.01, 0.01, 0.1, 0.2, 0.001, 0.00001])
y = torch.tensor([0, 0, 0, 1, 0, 0])
deviance(yhat, y)
>tensor(0.3101)

def tweedie_deviance_score(yhat, y):
  return 2 * (y * torch.log(y/yhat) + yhat - y)

tweedie_deviance_score(yhat, y)
>tensor([   nan,    nan,    nan, 1.6189,    nan,    nan])

As far as I can tell the formula in my function is the same as what is stated in the docs for power=1 (loss Poisson distribution). What am I missing here?