Rolv-Arild commited on
Commit
81966e1
·
verified ·
1 Parent(s): 3d1042f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -177,7 +177,7 @@ def plot_plotly(preds: pd.DataFrame):
177
  # Add touch indicators as points
178
  touches = {}
179
  for touch_frame in preds.index[preds["Touch"] != ""]:
180
- teams_players = preds["Touch"][touch_frame].split('|')
181
  for team, player in zip(teams_players[::2], teams_players[1::2]):
182
  team = team.strip()
183
  player = player.strip()
@@ -296,6 +296,7 @@ with TemporaryDirectory() as temp_dir:
296
  if os.path.exists(preds_file):
297
  print(f"Predictions file already exists: {preds_file}")
298
  preds = pd.read_csv(preds_file, dtype={"Touch": str})
 
299
  else:
300
  preds = infer(MODEL, replay_file,
301
  nullify_goal_difference=nullify_goal_difference,
 
177
  # Add touch indicators as points
178
  touches = {}
179
  for touch_frame in preds.index[preds["Touch"] != ""]:
180
+ teams_players = preds.at[touch_frame, "Touch"].split('|')
181
  for team, player in zip(teams_players[::2], teams_players[1::2]):
182
  team = team.strip()
183
  player = player.strip()
 
296
  if os.path.exists(preds_file):
297
  print(f"Predictions file already exists: {preds_file}")
298
  preds = pd.read_csv(preds_file, dtype={"Touch": str})
299
+ preds["Touch"] = preds["Touch"].fillna("")
300
  else:
301
  preds = infer(MODEL, replay_file,
302
  nullify_goal_difference=nullify_goal_difference,