TheXeos commited on
Commit
44c646a
1 Parent(s): 4b1b8a4

Fix some naming again

Browse files
Files changed (1) hide show
  1. background_task.py +2 -2
background_task.py CHANGED
@@ -96,7 +96,7 @@ class Matchmaking:
96
 
97
  def to_csv(self):
98
  """ Save the match history as a CSV file to the hub. """
99
- data_dict = {"author": [], "name": [], "elo": [], "games_played": []}
100
  for model in self.models:
101
  data_dict["author"].append(model.author)
102
  data_dict["model"].append(model.name)
@@ -140,7 +140,7 @@ def get_models_list() -> list:
140
  models_on_hub = []
141
  for i, row in data.iterrows():
142
  models.append(Model(row["author"], row["model"], row["elo"], row["games_played"]))
143
- models_names.append(row["name"])
144
  for model in models_on_hub:
145
  if model.modelId not in models_names:
146
  models.append(Model(model.author, model.modelId))
 
96
 
97
  def to_csv(self):
98
  """ Save the match history as a CSV file to the hub. """
99
+ data_dict = {"author": [], "model": [], "elo": [], "games_played": []}
100
  for model in self.models:
101
  data_dict["author"].append(model.author)
102
  data_dict["model"].append(model.name)
 
140
  models_on_hub = []
141
  for i, row in data.iterrows():
142
  models.append(Model(row["author"], row["model"], row["elo"], row["games_played"]))
143
+ models_names.append(row["model"])
144
  for model in models_on_hub:
145
  if model.modelId not in models_names:
146
  models.append(Model(model.author, model.modelId))