Allow to include in progress week

src/analyze_yahoo.py

@@ -2,11 +2,14 @@ import pandas as pd
 from typing import Optional
 
 
-def calculate_luck(df: pd.DataFrame, as_of_week: Optional[int] = None) -> pd.DataFrame:
+def calculate_luck(df: pd.DataFrame, as_of_week: Optional[int] = None, include_current: bool = False) -> pd.DataFrame:
     if as_of_week:
         df_complete = df[df.week <= as_of_week]
     else:
-        df_complete = df[df.matchup_status == "postevent"]
+        status_list = ["postevent"]
+        if include_current:
+            status_list.append("midevent")
+        df_complete = df[df.matchup_status.isin(status_list)]
     df_complete["actual_wins"] = df_complete["win_probability"].apply(lambda x: x > 0.5)
     df_list = []
     n_teams = df.team_name.nunique()

src/pages/51_League_Results_Summary.py

@@ -12,9 +12,9 @@ def get_all_league_settings_with_cache(season: int):
 
 
 @st.cache_resource(ttl=60 * 10)
-def get_cached_league_luck(league_key):
+def get_cached_league_luck(league_key, include_current: bool):
     df_weekly_results = st.session_state.yahoo_client.full_schedule_dataframe(league_key)
-    df_weekly_luck = calculate_luck(df_weekly_results)
+    df_weekly_luck = calculate_luck(df_weekly_results, include_current=include_current)
     df_grouped_luck = get_grouped_luck(df_weekly_luck)
     return df_weekly_results, df_weekly_luck, df_grouped_luck
 
@@ -37,7 +37,8 @@ def get_page():
         selected_league = st.selectbox("Select league", user_leagues, format_func=lambda x: x.name)
         st.header(f"{selected_league.name} - {selected_league.season}")
 
-        df_weekly_results, df_weekly_luck, df_luck = get_cached_league_luck(selected_league.league_key)
+        check_current = st.checkbox("Include week in progress")
+        df_weekly_results, df_weekly_luck, df_luck = get_cached_league_luck(selected_league.league_key, check_current)
 
         st.header("Weekly Top Half Scores")
         luck_pivot = df_weekly_luck.pivot_table(