Check if model is none

background_task.py

@@ -7,13 +7,11 @@ import pandas as pd
 from datetime import datetime
 from huggingface_hub import HfApi, Repository
 
-
 DATASET_REPO_URL = "https://huggingface.co/datasets/CarlCochet/BotFightData"
 ELO_FILENAME = "soccer_elo.csv"
 ELO_DIR = "soccer_elo"
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
-
 subprocess.run("rm -rf .git/hooks".split())
 repo = Repository(
     local_dir=ELO_DIR, clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
@@ -30,6 +28,7 @@ class Model:
     :param elo: Elo rating of the model
     :param games_played: Number of games played by the model (useful if we implement sigma uncertainty)
     """
+
     def __init__(self, author, name, elo=1200, games_played=0):
         self.author = author
         self.name = name
@@ -47,6 +46,7 @@ class Matchmaking:
     :param max_diff: Maximum difference considered between two models' elo
     :param matches: Dictionary containing the match history (to later upload as CSV)
     """
+
     def __init__(self, models):
         self.models = models
         self.queue = self.models.copy()
@@ -92,11 +92,12 @@ class Matchmaking:
             model1 = self.find_model(model1[0], model1[1])
             model2 = self.find_model(model2[0], model2[1])
             result = row["result"]
-            self.compute_elo(row["model1"], row["model2"], row["result"])
-            self.matches["model1"].append(model1.name)
-            self.matches["model2"].append(model2.name)
-            self.matches["result"].append(result)
-            self.matches["timestamp"].append(row["timestamp"])
+            if model1 is not None or model2 is not None:
+                self.compute_elo(row["model1"], row["model2"], row["result"])
+                self.matches["model1"].append(model1.name)
+                self.matches["model2"].append(model2.name)
+                self.matches["result"].append(result)
+                self.matches["timestamp"].append(row["timestamp"])
 
     def find_model(self, author, name):
         """ Find a model in the models list. """