update

arena_elo/elo_rating/upload_battle_data.py

@@ -75,7 +75,7 @@ def main(
     task_type='t2i_generation',
     # data_file: str = "./results/latest/clean_battle_conv.json",
     data_file: str = None,
-    repo_id: str = "DongfuJiang/GenAI-Arena-human-eval",
+    repo_id: str = "TIGER-Lab/GenAI-Arena-human-eval",
     log_dir: str = os.getenv("LOGDIR", "../GenAI-Arena-hf-logs/vote_log"),
     config_name='battle',
     split='test',
@@ -92,7 +92,6 @@ def main(
     # add index according to the tsamp
     if seed is not None:
         random.seed(seed)
-    # data = random.sample(data, 50 * 7)
     
     
     data = sorted(data, key=lambda x: x['tstamp'])
@@ -104,11 +103,15 @@ def main(
     valid_data = []
     for i, battle in enumerate(data):
         if any(key not in battle['inputs'] for key in required_keys_each_task[task_type]):
+            # print(battle['inputs'])
             # print(f"Skipping battle {i} due to missing keys")
             continue
         valid_data.append(battle)
+    print(f"Total battles: {len(data)}, valid battles: {len(valid_data)}, removed battles: {len(data) - len(valid_data)}")
     data = valid_data
     
+    # data = random.sample(data, 50 * 7+2)
+    
     for i, battle in enumerate(data):
         battle['index'] = i