James McCool commited on
Commit
b3a8168
·
1 Parent(s): d6b4eb2

Update team data processing in init_player_data function to convert unique team names to a list. This change improves compatibility with subsequent database queries, enhancing the accuracy of team statistics retrieval during simulations.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -391,7 +391,7 @@ def init_player_data(game_count, players, opponent, win_loss_settings, kill_pred
391
  cursor = collection.find({"playername": {"$in": players}, "date": {"$gte": start_datetime, "$lte": end_datetime}})
392
  raw_display = pd.DataFrame(list(cursor))
393
 
394
- teams = raw_display['teamname'].unique()
395
 
396
  cursor = collection.find({"teamname": {"$in": teams}, "date": {"$gte": start_datetime, "$lte": end_datetime}})
397
  raw_team = pd.DataFrame(list(cursor))
 
391
  cursor = collection.find({"playername": {"$in": players}, "date": {"$gte": start_datetime, "$lte": end_datetime}})
392
  raw_display = pd.DataFrame(list(cursor))
393
 
394
+ teams = raw_display['teamname'].unique().tolist()
395
 
396
  cursor = collection.find({"teamname": {"$in": teams}, "date": {"$gte": start_datetime, "$lte": end_datetime}})
397
  raw_team = pd.DataFrame(list(cursor))