Update chess-gpt-eval/main.py

chess-gpt-eval/main.py

@@ -667,13 +667,19 @@ if __name__ == "__main__":
 print("\n\n\n********\nFinal Statistics:\n********\n")
 for nanogpt_player in player_ones:
     player_one_recording_name = "xformer_" + nanogpt_player
-    csv_file_path = f"logs/{player_one_recording_name}_vs_{player_two_recording_name}"
-    csv_file_path = csv_file_path.replace(".", "_") # Because I'm using ckpt filenames for nanogpt models 
-    csv_file_path += ".csv"
+    if RUN_FOR_ANALYSIS:
+        csv_file_path = f"logs/{player_one_recording_name}_vs_{player_two_recording_name}"
+        csv_file_path = csv_file_path.replace(".", "_") # Because I'm using ckpt filenames for nanogpt models 
+        csv_file_path += ".csv"
+    else:
+        csv_file_path = recording_file
     
-    stats = calculate_stats(csv_file_path)
-    if stats:
-        print(f"\nStatistics for {nanogpt_player}:")
-        for key, value in stats.items():
-            print(f"{key}: {value}")
+    try:
+        stats = calculate_stats(csv_file_path)
+        if stats:
+            print(f"\nStatistics for {nanogpt_player}:")
+            for key, value in stats.items():
+                print(f"{key}: {value}")
+    except:
+        print(f"Couldn't get stats for {csv_file_path}")
 print("\n\n\n********\nDONE!\n********\n\n\n")