WE GOIN

app.py

@@ -9,13 +9,13 @@ from feature_engineering import heroes, hero_information
 
 # Define expected columns
 EXPECTED_COLUMNS = ['mmr', 'p1', 'p2', 'p3', 'p4', 'p5', 'count', 'mean', 'std', 'min', 'max', 'sum',
-                   'total_games_played', 'total_winrate'] + \
-                  [f'games_{i}' for i in range(1, 146) if i != 24 and i != 122 and i != 124 and i != 125 and 
-                   i != 127 and i != 130 and i != 132 and i != 133 and i != 134 and i != 139 and 
-                   i != 140 and i != 141 and i != 142 and i != 143 and i != 144] + \
-                  [f'winrate_{i}' for i in range(1, 146) if i != 24 and i != 122 and i != 124 and i != 125 and 
-                   i != 127 and i != 130 and i != 132 and i != 133 and i != 134 and i != 139 and 
-                   i != 140 and i != 141 and i != 142 and i != 143 and i != 144]
+                   'total_games_played', 'total_winrate']
+
+# Add games columns
+games_ids = ([i for i in range(1, 115)] +  # 1-114
+            [119, 120, 121, 123, 126, 128, 129, 131, 135, 136, 137, 138, 145])
+EXPECTED_COLUMNS.extend([f'games_{i}' for i in games_ids])
+EXPECTED_COLUMNS.extend([f'winrate_{i}' for i in games_ids])
 
 def prepare_single_player_data(user_id, mmr, comf_1, comf_2, comf_3, comf_4, comf_5):
     """Creates a DataFrame in the expected format for the model"""