Jon Solow commited on
Commit
b4d2f11
1 Parent(s): 72c8544

Fix override logic to allow setting value to zero and drop all nans

Browse files
Files changed (1) hide show
  1. src/login.py +2 -4
src/login.py CHANGED
@@ -163,13 +163,11 @@ def get_stat_overrides() -> dict[int, dict[str, dict[str, float]]]:
163
  usecols=[0, 1, 2, 3],
164
  )
165
  stat_overrides_map: dict[int, dict[str, dict[str, float]]] = {w: {} for w in df.week.values}
166
-
167
- for week, df_week in df.groupby("week"):
168
  for player_id, df_player in df_week.groupby("player_id"):
169
  if player_id not in stat_overrides_map[week]:
170
  stat_overrides_map[week][player_id] = {}
171
  for row in df_player.itertuples():
172
- if isinstance(row.stat_key, str) and row.stat_value:
173
  stat_overrides_map[week][player_id][row.stat_key] = float(row.stat_value)
174
-
175
  return stat_overrides_map
 
163
  usecols=[0, 1, 2, 3],
164
  )
165
  stat_overrides_map: dict[int, dict[str, dict[str, float]]] = {w: {} for w in df.week.values}
166
+ for week, df_week in df.dropna().groupby("week"):
 
167
  for player_id, df_player in df_week.groupby("player_id"):
168
  if player_id not in stat_overrides_map[week]:
169
  stat_overrides_map[week][player_id] = {}
170
  for row in df_player.itertuples():
171
+ if isinstance(row.stat_key, str):
172
  stat_overrides_map[week][player_id][row.stat_key] = float(row.stat_value)
 
173
  return stat_overrides_map