Spaces:
Runtime error
Runtime error
Fix some naming
Browse files- background_task.py +5 -5
- 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["
|
| 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["
|
| 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
|