Jon Solow commited on
Commit
13801e8
·
1 Parent(s): 7fd5387

Implement query for league schedule from pfr

Browse files
Files changed (1) hide show
  1. src/queries/pfr/league_schedule.py +2 -2
src/queries/pfr/league_schedule.py CHANGED
@@ -10,7 +10,7 @@ def get_full_schedule(season_int: str | int) -> pd.DataFrame:
10
  return df
11
 
12
 
13
- def get_week_team_time_map(df_schedule: pd.DataFrame) -> dict[int, dict[str, str]]:
14
  min_week = 1
15
  max_week = 23
16
  week_team_time_map: dict[int, dict[str, str]] = {k: {} for k in range(min_week, max_week + 1)}
@@ -21,6 +21,6 @@ def get_week_team_time_map(df_schedule: pd.DataFrame) -> dict[int, dict[str, str
21
  return week_team_time_map
22
 
23
 
24
- def get_season_time_map(season_int: str | int) -> dict[int, dict[str, str]]:
25
  df_schedule = get_full_schedule(season_int)
26
  return get_week_team_time_map(df_schedule)
 
10
  return df
11
 
12
 
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, str]] = {k: {} for k in range(min_week, max_week + 1)}
 
21
  return week_team_time_map
22
 
23
 
24
+ def get_season_time_map(season_int: str | int) -> dict[int, dict[str, pd.Timestamp]]:
25
  df_schedule = get_full_schedule(season_int)
26
  return get_week_team_time_map(df_schedule)