From 271a678e73216ee252cc1af0d6eaf9ed76b9596a Mon Sep 17 00:00:00 2001
From: Jiarui Li <jiarui_li@outlook.com>
Date: Mon, 3 Aug 2026 08:33:21 +0800
Subject: [PATCH] Fix calibration death token lookup

---
 evaluate_calibration.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/evaluate_calibration.py b/evaluate_calibration.py
index 6422703..73eb210 100644
--- a/evaluate_calibration.py
+++ b/evaluate_calibration.py
@@ -1755,7 +1755,7 @@ def main() -> None:
     if landmark_ages.size == 0:
         raise ValueError("No landmark ages selected.")
 
-    death_token_ids = _get_death_token_ids(dataset, labels_meta)
+    death_token_ids = _get_death_token_ids(dataset)
     landmark_dataset = LandmarkDataset(
         dataset=dataset,
         subset_indices=subset_indices,