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- 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)
|
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
|