Add loading / sending to hub

main.py

@@ -1,6 +1,7 @@
 import random
 import pandas as pd
 from datetime import datetime
+from huggingface_hub import hf_hub_download, HfApi
 
 
 class Model:
@@ -37,7 +38,8 @@ class Matchmaking:
             "model1": [],
             "model2": [],
             "result": [],
-            "datetime": []
+            "datetime": [],
+            "env": []
         }
 
     def run(self):
@@ -58,6 +60,7 @@ class Matchmaking:
                 self.matches["model2"].append(model2.name)
                 self.matches["result"].append(result)
                 self.matches["datetime"].append(datetime.now().strftime("%Y-%m-%d %H:%M:%S.%f"))
+                self.matches["env"].append(self.env)
 
     def compute_elo(self, model1, model2, result):
         """ Compute the new elo for each model based on a match result. """
@@ -96,7 +99,13 @@ class Matchmaking:
             df = pd.concat([df, pd.DataFrame([[model.name, model.elo]], columns=['name', 'elo'])])
         df.to_csv('elo.csv', index=False)
         df_matches = pd.DataFrame(self.matches)
-        df_matches.to_csv(f"matches/{self.env}__{datetime.now().strftime('%Y-%m-%d_%H-%M-%S_%f')}.csv", index=False)
+        date = datetime.now()
+        df_matches.to_csv(f"matches/{self.env}__{date.strftime('%Y-%m-%d_%H-%M-%S_%f')}.csv", index=False)
+        api.upload_file(
+            path_or_fileobj=f"matches/{self.env}__{date.strftime('%Y-%m-%d_%H-%M-%S_%f')}.csv",
+            path_in_repo=f"match_history/{self.env}__{date.strftime('%Y-%m-%d_%H-%M-%S_%f')}.csv",
+            repo_id="CarlCochet/BotFights"
+        )
 
 
 def match(model1, model2) -> float:
@@ -120,7 +129,11 @@ def get_models_list() -> list:
     :return: list of Model objects
     """
     models = []
-    data = pd.read_csv("example.csv")
+    hf_hub_download(
+        repo_id="CarlCochet/BotFights",
+        filename="elo.csv",
+    )
+    data = pd.read_csv("elo.csv")
     for i, row in data.iterrows():
         models.append(Model(row["name"], row["elo"]))
     return models
@@ -128,11 +141,12 @@ def get_models_list() -> list:
 
 def init_matchmaking():
     models = get_models_list()
-    matchmaking = Matchmaking(models)
+    matchmaking = Matchmaking(models, "snowball-fight")
     matchmaking.run()
     matchmaking.to_csv()
 
 
 if __name__ == "__main__":
     print("It's running!")
-    # init_matchmaking()
+    api = HfApi()
+    init_matchmaking()