Skip to content

Commit ad9fec4

Browse files
committed
allow batched implementation of censored and truncated distributions
1 parent b73f506 commit ad9fec4

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

scoringrules/core/crps/_closed.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -316,8 +316,8 @@ def gtclogistic(
316316

317317
c = (1 - lmass - umass) / (F_u - F_l)
318318

319-
s1_u = B.where(u_inf and umass == 0.0, 0.0, u * umass**2)
320-
s1_l = B.where(l_inf and lmass == 0.0, 0.0, l * lmass**2)
319+
s1_u = B.where(u_inf & (umass == 0.0), 0.0, u * umass**2)
320+
s1_l = B.where(l_inf & (lmass == 0.0), 0.0, l * lmass**2)
321321

322322
s1 = B.abs(ω - z) + s1_u - s1_l
323323
s2 = c * z * ((1 - 2 * lmass) * F_u + (1 - 2 * umass) * F_l) / (1 - lmass - umass)
@@ -359,8 +359,8 @@ def gtcnormal(
359359

360360
u = B.where(u_inf, B.nan, u)
361361
l = B.where(l_inf, B.nan, l)
362-
s1_u = B.where(u_inf and umass == 0.0, 0.0, u * umass**2)
363-
s1_l = B.where(l_inf and lmass == 0.0, 0.0, l * lmass**2)
362+
s1_u = B.where(u_inf & (umass == 0.0), 0.0, u * umass**2)
363+
s1_l = B.where(l_inf & (lmass == 0.0), 0.0, l * lmass**2)
364364

365365
c = (1 - lmass - umass) / (F_u - F_l)
366366

@@ -410,8 +410,8 @@ def gtct(
410410
u = B.where(u_inf, B.nan, u)
411411
l = B.where(l_inf, B.nan, l)
412412

413-
s1_u = B.where(u_inf and umass == 0.0, 0.0, u * umass**2)
414-
s1_l = B.where(l_inf and lmass == 0.0, 0.0, l * lmass**2)
413+
s1_u = B.where(u_inf & (umass == 0.0), 0.0, u * umass**2)
414+
s1_l = B.where(l_inf & (lmass == 0.0), 0.0, l * lmass**2)
415415

416416
G_u = B.where(u_inf, 0.0, -f_u * (df + u**2) / (df - 1))
417417
G_l = B.where(l_inf, 0.0, -f_l * (df + l**2) / (df - 1))

0 commit comments

Comments
 (0)