From 00922cb15abc46299b1e3e287be692289231907d Mon Sep 17 00:00:00 2001 From: Tianqi Liu Date: Mon, 22 Jan 2024 14:03:51 -0800 Subject: [PATCH] internal_change PiperOrigin-RevId: 600564077 --- rax/utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/rax/utils.py b/rax/utils.py index e4d74a6..4fd1859 100644 --- a/rax/utils.py +++ b/rax/utils.py @@ -14,6 +14,7 @@ """Rax-specific utilities.""" +from rax._src.losses import pairwise_loss from rax._src.utils import approx_cutoff from rax._src.utils import approx_ranks from rax._src.utils import compute_pairs @@ -25,5 +26,6 @@ "approx_ranks", "compute_pairs", "cutoff", + "pairwise_loss", "ranks", ]