Update app.py

app.py

@@ -120,6 +120,7 @@ class DAggerEncoder:
         })
         self.outputs.append(label)
         worker_status["records_extracted"] += 1
+        print(worker_status["records_extracted"])
         
         if len(self.inputs) >= self.chunk_size:
             self.save_and_upload()
@@ -205,6 +206,19 @@ def play_dagger_game(encoder: DAggerEncoder, student_bots: dict, teacher_bots: d
             if student_final_action is None:
                 student_final_action = env_obs.select_action_from_mjai('{"type": "none"}')
                 
+            # 兜底 2: 如果不能 Pass（比如自己摸牌后必须切牌），强制选取合法动作
+            if student_final_action is None:
+                try:
+                    # 获取当前环境允许的所有合法动作
+                    legal_actions = env_obs.legal_actions
+                    if callable(legal_actions):
+                        legal_actions = legal_actions()
+                        
+                    if legal_actions and len(legal_actions) > 0:
+                        student_final_action = legal_actions[0] # 强行拿第一个合法动作兜底
+                except Exception as e:
+                    pass
+                    
             actions_to_env[pid] = student_final_action
 
         # 环境步进