Jon Solow
commited on
Commit
·
c762be5
1
Parent(s):
14ccd13
Fix typehint
Browse files
src/queries/pfr/league_schedule.py
CHANGED
@@ -13,7 +13,7 @@ def get_full_schedule(season_int: str | int) -> pd.DataFrame:
|
|
13 |
def get_week_team_time_map(df_schedule: pd.DataFrame) -> dict[int, dict[str, pd.Timestamp]]:
|
14 |
min_week = 1
|
15 |
max_week = 23
|
16 |
-
week_team_time_map: dict[int, dict[str,
|
17 |
for _, row in df_schedule.iterrows():
|
18 |
game_time = pd.to_datetime(row.Date + " " + row.Time, yearfirst=True)
|
19 |
week_team_time_map[int(row.Week)][row["Winner/tie"]] = game_time
|
|
|
13 |
def get_week_team_time_map(df_schedule: pd.DataFrame) -> dict[int, dict[str, pd.Timestamp]]:
|
14 |
min_week = 1
|
15 |
max_week = 23
|
16 |
+
week_team_time_map: dict[int, dict[str, pd.Timestamp]] = {k: {} for k in range(min_week, max_week + 1)}
|
17 |
for _, row in df_schedule.iterrows():
|
18 |
game_time = pd.to_datetime(row.Date + " " + row.Time, yearfirst=True)
|
19 |
week_team_time_map[int(row.Week)][row["Winner/tie"]] = game_time
|