Update eval.py (#11)

McGill-NLP · Mar 9, 2024 · bf1716c · bf1716c
1 parent 2c21ef6
commit bf1716c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modeling/dmr/eval.py b/modeling/dmr/eval.py
@@ -88,7 +88,7 @@ def run_model_and_update_groups(
             r["score"] = scores[i]
 
 
-def build_target_uids_dict(demos, uid_key="data-weblinx-id"):
+def build_target_uids_dict(demos, uid_key="data-webtasks-id"):
     """
     Given a list of demonstrations, build a dictionary mapping
     `(demo_name, turn_index) -> uid`. This is used to determine the