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
|