Enhance lineup processing in hedging_preset.py by adding checks for empty DataFrames before generating lineups. This update prevents errors when no lineups are available and improves the clarity of debug messages related to lineup generation.

global_func/hedging_preset.py

@@ -51,16 +51,19 @@ def hedging_preset(portfolio: pd.DataFrame, lineup_target: int, projections_file
         )
 
         removed_df = working_df[remove_mask]
-        print(removed_df.head(10))
         locked_df = working_df[lock_mask]
-        print(locked_df.head(10))
 
         removed_lineups = small_field_preset(removed_df, math.ceil(lineup_target / (list_size * 3)), excluded_cols)
         print(len(removed_lineups))
-        locked_lineups = large_field_preset(locked_df, math.ceil(lineup_target / (list_size * 3)), excluded_cols)
-        print(len(locked_lineups))
-
-        concat_portfolio = pd.concat([concat_portfolio, removed_lineups, locked_lineups])
+        # Check if locked_df is empty before calling large_field_preset
+        if not locked_df.empty:
+            locked_lineups = large_field_preset(locked_df, math.ceil(lineup_target / (list_size * 3)), excluded_cols)
+            print(len(locked_lineups))
+            concat_portfolio = pd.concat([concat_portfolio, removed_lineups, locked_lineups])
+        else:
+            # If no lineups contain this player, just add the removed lineups
+            print(f"No lineups found containing {player}")
+            concat_portfolio = pd.concat([concat_portfolio, removed_lineups])
     
     for team in top_owned_teams:
         working_df = portfolio.copy()
@@ -68,11 +71,13 @@ def hedging_preset(portfolio: pd.DataFrame, lineup_target: int, projections_file
         teams_df = working_df[working_df['Stack'] == team]
 
         removed_lineups = small_field_preset(removed_df, math.ceil(lineup_target / (list_size * 3)), excluded_cols)
-        print(len(removed_lineups))
-        print(removed_lineups.head(10))
-        team_lineups = large_field_preset(teams_df, math.ceil(lineup_target / (list_size * 3)), excluded_cols)
-        print(len(team_lineups))
-        print(team_lineups.head(10))
-        concat_portfolio = pd.concat([concat_portfolio, removed_lineups, team_lineups])
+        # Check if teams_df is empty before calling large_field_preset
+        if not teams_df.empty:
+            team_lineups = large_field_preset(teams_df, math.ceil(lineup_target / (list_size * 3)), excluded_cols)
+            concat_portfolio = pd.concat([concat_portfolio, removed_lineups, team_lineups])
+        else:
+            # If no lineups have this team stacked, just add the removed lineups
+            print(f"No lineups found with {team} stacked")
+            concat_portfolio = pd.concat([concat_portfolio, removed_lineups])
 
     return concat_portfolio.head(lineup_target)