TheXeos commited on
Commit
4b1b8a4
1 Parent(s): 6306e8a

Fix some naming

Browse files
Files changed (2) hide show
  1. background_task.py +5 -5
  2. match_history/.gitkeep +0 -0
background_task.py CHANGED
@@ -99,7 +99,7 @@ class Matchmaking:
99
  data_dict = {"author": [], "name": [], "elo": [], "games_played": []}
100
  for model in self.models:
101
  data_dict["author"].append(model.author)
102
- data_dict["name"].append(model.name)
103
  data_dict["elo"].append(model.elo)
104
  data_dict["games_played"].append(model.games_played)
105
 
@@ -139,7 +139,7 @@ def get_models_list() -> list:
139
  # models_on_hub = api.list_models(filter=["reinforcement-learning", env, "stable-baselines3"])
140
  models_on_hub = []
141
  for i, row in data.iterrows():
142
- models.append(Model(row["author"], row["name"], 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:
@@ -149,9 +149,9 @@ def get_models_list() -> list:
149
 
150
  def init_matchmaking():
151
  models = get_models_list()
152
- matchmaking = Matchmaking(models)
153
- matchmaking.run()
154
- matchmaking.to_csv()
155
 
156
 
157
  if __name__ == "__main__":
 
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)
103
  data_dict["elo"].append(model.elo)
104
  data_dict["games_played"].append(model.games_played)
105
 
 
139
  # models_on_hub = api.list_models(filter=["reinforcement-learning", env, "stable-baselines3"])
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:
 
149
 
150
  def init_matchmaking():
151
  models = get_models_list()
152
+ # matchmaking = Matchmaking(models)
153
+ # matchmaking.run()
154
+ # matchmaking.to_csv()
155
 
156
 
157
  if __name__ == "__main__":
match_history/.gitkeep ADDED
File without changes