Spaces:
Runtime error
Runtime error
Add back matchmaking process
Browse files- background_task.py +8 -5
background_task.py
CHANGED
@@ -96,8 +96,10 @@ class Matchmaking:
|
|
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 |
-
|
|
|
|
|
101 |
data_dict["author"].append(model.author)
|
102 |
data_dict["model"].append(model.name)
|
103 |
data_dict["elo"].append(model.elo)
|
@@ -149,9 +151,10 @@ def get_models_list() -> list:
|
|
149 |
|
150 |
def init_matchmaking():
|
151 |
models = get_models_list()
|
152 |
-
|
153 |
-
|
154 |
-
|
|
|
155 |
|
156 |
|
157 |
if __name__ == "__main__":
|
|
|
96 |
|
97 |
def to_csv(self):
|
98 |
""" Save the match history as a CSV file to the hub. """
|
99 |
+
data_dict = {"rank": [], "author": [], "model": [], "elo": [], "games_played": []}
|
100 |
+
sorted_models = sorted(self.models, key=lambda x: x.elo, reverse=True)
|
101 |
+
for i, model in enumerate(sorted_models):
|
102 |
+
data_dict["rank"].append(i + 1)
|
103 |
data_dict["author"].append(model.author)
|
104 |
data_dict["model"].append(model.name)
|
105 |
data_dict["elo"].append(model.elo)
|
|
|
151 |
|
152 |
def init_matchmaking():
|
153 |
models = get_models_list()
|
154 |
+
matchmaking = Matchmaking(models)
|
155 |
+
matchmaking.run()
|
156 |
+
matchmaking.to_csv()
|
157 |
+
print("Matchmaking done ---", datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f"))
|
158 |
|
159 |
|
160 |
if __name__ == "__main__":
|