modified: losses.py

autonlab · Apr 13, 2021 · 2801c6c · 2801c6c
1 parent 7a968a8
commit 2801c6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dsm/losses.py b/dsm/losses.py
@@ -326,7 +326,7 @@ def _weibull_pdf(model, x, t_horizon, risk='1'):
     lpdfs = torch.logsumexp(lpdfs, dim=1)
     pdfs.append(lpdfs.detach().numpy())
 
-  return lpdfs
+  return pdfs
 
 def _weibull_cdf(model, x, t_horizon, risk='1'):