Fix calibration death token lookup

2026-08-03 08:33:21 +08:00
parent 4f102fb271
commit 271a678e73
1 changed files with 1 additions and 1 deletions
--- 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,