No public description

PiperOrigin-RevId: 597955531
google · Jan 12, 2024 · 6ab3524 · 6ab3524
1 parent 1a05bb3
commit 6ab3524
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/baselines/t5/data/metrics.py b/baselines/t5/data/metrics.py
@@ -112,8 +112,8 @@ def wrapped_seq2seq_uncertainty_metrics(
 
 
 def _seq2seq_uncertainty_metrics(
-    targets: List[Text],
-    predictions: List[Text],
+    targets: List[Text | int],
+    predictions: List[Text | int],
     log_probs: np.ndarray,
     num_ece_bins: int = 15,
     metric_prefix: str = '',