Update elo.py

elo.py

@@ -1,13 +1,15 @@
 import pandas as pd
 
-def update_elo_ratings(ratings_dataset, winner, loser):
-    # Check if the ratings_dataset is empty
-    if ratings_dataset is None or ratings_dataset.num_rows == 0:
+def update_elo_ratings(ratings_dict, winner, loser):
+    # Check if the ratings_dict is empty
+    if not ratings_dict:
         # Create a blank DataFrame with the required columns
         ratings_df = pd.DataFrame(columns=['bot_name', 'elo_rating', 'games_played'])
     else:
-        # Convert the Hugging Face dataset to a pandas DataFrame
-        ratings_df = pd.DataFrame(ratings_dataset)
+        # Convert the dictionary to a pandas DataFrame
+        ratings_df = pd.DataFrame.from_dict(ratings_dict, orient='index')
+        ratings_df.reset_index(inplace=True)
+        ratings_df.columns = ['bot_name', 'elo_rating', 'games_played']
 
     # Check and add new players if they don't exist in the dataset
     for player in [winner, loser]:
@@ -46,7 +48,7 @@ def update_elo_ratings(ratings_dataset, winner, loser):
     ratings_df.loc[ratings_df['bot_name'] == winner, 'elo_rating'] = winner_new_rating
     ratings_df.loc[ratings_df['bot_name'] == loser, 'elo_rating'] = loser_new_rating
 
-    # Convert the DataFrame back to a Hugging Face dataset
-    updated_ratings_dataset = Dataset.from_pandas(ratings_df)
+    # Convert the DataFrame to a dictionary
+    updated_ratings_dict = ratings_df.set_index('bot_name').to_dict(orient='index')
 
-    return updated_ratings_dataset
+    return updated_ratings_dict