Upload chess-gpt-eval/mamba_module.py with huggingface_hub

chess-gpt-eval/mamba_module.py

@@ -127,14 +127,17 @@ class MambaPlayer:
      #   return ''.join([itos[id] for id in token_ids])  # Example
         
     def get_move_from_response(self, response: str) -> str:
-        if not response:
+        if not response or len(response) == 0:
             return None
         # Parse the response to get only the first move
-        moves = response.split()
-        first_move = moves[0]
-        first_move = first_move.lstrip('.') # A patch for a weird phase during training ... doesn't seem to be an issue anymore, but don't see the harm.
+        try:
+            moves = response.split()
+            first_move = moves[0]
+            first_move = first_move.lstrip('.') # A patch for a weird phase during training ... doesn't seem to be an issue anymore, but don't see the harm.
 
-        return first_move
+            return first_move
+        except:
+            return None
 
     def get_move(self, board: str, game_state: str, temperature: float) -> str:
         completion = self.get_mamba_response(game_state, temperature, 8, self.vocab_size)