fix

SumanthRH · SumanthRH · commit 8e3daade5eac · 2025-02-20T13:03:27.000-08:00
Signed-off-by: SumanthRH &lt;sumanthrh@anyscale.com&gt;
diff --git a/skythought/skythought_evals/tasks/base.py b/skythought/skythought_evals/tasks/base.py
@@ -102,7 +102,7 @@ def process_remaining_data(self, train_data, id_to_results: dict):
         return [
             row.to_dict()
             for _, row in train_data.iterrows()
-            if int(row["_index"]) not in id_to_results
+            if str(row["_index"]) not in id_to_results
         ]
 
 

Original file line number	Diff line number	Diff line change
`@@ -102,7 +102,7 @@ def process_remaining_data(self, train_data, id_to_results: dict):`
`102`	`102`	`return [`
`103`	`103`	`row.to_dict()`
`104`	`104`	`for _, row in train_data.iterrows()`
`105`		`- if int(row["_index"]) not in id_to_results`
	`105`	`+ if str(row["_index"]) not in id_to_results`
`106`	`106`	`]`
`107`	`107`
`108`	`108`