Update elo.py

elo.py

@@ -29,16 +29,16 @@ def update_elo_ratings(ratings_dataset, winner, loser, k_factor=32):
     ratings_df = pd.DataFrame(ratings_dataset)
 
     # Extract old ratings
-    winner_old_rating = ratings_df.loc[ratings_df == winner, 'rating'].iloc[0]
-    loser_old_rating = ratings_df.loc[ratings_df == loser, 'rating'].iloc[0]
+    winner_old_rating = ratings_df.loc[ratings_df == winner, 'elo_rating'].iloc[0]
+    loser_old_rating = ratings_df.loc[ratings_df == loser, 'elo_rating'].iloc[0]
 
     # Calculate new ratings
     winner_new_rating = calculate_elo(winner_old_rating, loser_old_rating, 1, k_factor)
     loser_new_rating = calculate_elo(loser_old_rating, winner_old_rating, 0, k_factor)
 
     # Update the DataFrame
-    ratings_df.loc[ratings_df == winner, 'rating'] = winner_new_rating
-    ratings_df.loc[ratings_df == loser, 'rating'] = loser_new_rating
+    ratings_df.loc[ratings_df == winner, 'elo_rating'] = winner_new_rating
+    ratings_df.loc[ratings_df == loser, 'elo_rating'] = loser_new_rating
 
     # Convert the DataFrame back to a Hugging Face dataset
     updated_ratings_dataset = Dataset.from_pandas(ratings_df)