Update chess-gpt-eval-contrastive/main.py

chess-gpt-eval-contrastive/main.py

@@ -629,24 +629,24 @@ def play_games(
         if update_activations:
             if player_one_resignation or player_one_failed_to_find_legal_move:
                 player_one.update_activations("lost")
-                player_two.update_activations("won")
+                #player_two.update_activations("won")
             elif player_two_resignation or player_two_failed_to_find_legal_move:
                 player_one.update_activations("won")
-                player_two.update_activations("lost")
+                #player_two.update_activations("lost")
             else:
                 if board.result() == "1-0":
                     player_one.update_activations("won")
-                    player_two.update_activations("lost")
+                 #   player_two.update_activations("lost")
                 elif board.result() == "0-1":
                     player_one.update_activations("lost")
-                    player_two.update_activations("won")
+                  #  player_two.update_activations("won")
                 else:  # Draw
                     player_one.update_activations("draw")
-                    player_two.update_activations("draw")
+                   # player_two.update_activations("draw")
         
-        if games_saved % contrastive_activation_save_interval == 0:
+        if update_activations and games_saved % contrastive_activation_save_interval == 0:
             player_one.save_activations(activations_path)
-            player_two.save_activations(activations_path)
+            #player_two.save_activations(activations_path)
             
     if isinstance(player_one, StockfishPlayer):
         player_one.close()