Jon Solow commited on
Commit
b3bd1da
·
1 Parent(s): 2eee45f

Fix type hint to allow int or str representation of season

Browse files
src/queries/nflverse/github_data.py CHANGED
@@ -20,7 +20,7 @@ FANTASY_POSITIONS = [
20
  ]
21
 
22
 
23
- def get_weekly_rosters(season_int: int = SEASON) -> pd.DataFrame:
24
  df = duckdb.sql(
25
  f"""
26
  with grouped_stats as (
@@ -46,7 +46,7 @@ def get_weekly_rosters(season_int: int = SEASON) -> pd.DataFrame:
46
  return df
47
 
48
 
49
- def get_player_stats(season_int: int = SEASON) -> pd.DataFrame:
50
  df = duckdb.sql(f"SELECT * from player_stats_player_stats_{season_int}").df()
51
  return df
52
 
 
20
  ]
21
 
22
 
23
+ def get_weekly_rosters(season_int: int | str = SEASON) -> pd.DataFrame:
24
  df = duckdb.sql(
25
  f"""
26
  with grouped_stats as (
 
46
  return df
47
 
48
 
49
+ def get_player_stats(season_int: int | str = SEASON) -> pd.DataFrame:
50
  df = duckdb.sql(f"SELECT * from player_stats_player_stats_{season_int}").df()
51
  return df
52