Add base file

env_elos/elo.csv

@@ -0,0 +1,3 @@
+model,elo,env
+model1,1200,snowballfight
+,,

matchmaking.py

@@ -11,10 +11,11 @@ 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, name, elo):
+    def __init__(self, name, elo, env):
         self.name = name
         self.elo = elo
         self.games_played = 0
+        self.env = env
 
 
 class Matchmaking:
@@ -46,7 +47,7 @@ class Matchmaking:
         self.matches = self.matches.sort_values("datetime")
         self.matches.reset_index(drop=True, inplace=True)
         model_names = self.matches["model1"].unique()
-        self.models = [Model(name, self.start_elo) for name in model_names]
+        self.models = [Model(name, self.start_elo, self.matches[self.matches["name"] == name].iloc[0]["env"]) for name in model_names]
 
     def compute_elo(self):
         """ Compute the elo for each model after each match. """