Update src/sports/nhl_functions.py

src/sports/nhl_functions.py

@@ -124,7 +124,7 @@ def run_go_classic_lineup_generator(site="DK", sport="NHL"):
         st.write(f"Unexpected error: {e}")
         return False
 
-def init_team_results(model_source: DataFrame, position_reqs: dict, salary_cap: int, max_skaters: int, teams_to_optimize: list, combos: int):
+def init_team_results(model_source: DataFrame, position_reqs: dict, salary_cap: int, max_skaters: int, teams_to_optimize: list, combos: int, player_team_map):
     df = model_source.copy()
     
     # Parse eligible positions for each player
@@ -531,7 +531,7 @@ def init_team_results(model_source: DataFrame, position_reqs: dict, salary_cap:
 
     return all_team_results
 
-def format_optimals(all_team_results: list, model_source: DataFrame, required_positions: list):
+def format_optimals(all_team_results: list, model_source: DataFrame, required_positions: list, player_team_map):
     roo_frame = model_source.copy()
 
     combined_df = pd_concat(all_team_results, ignore_index=True)
@@ -1875,8 +1875,8 @@ def DK_NHL_seed_frame(db, roo_file):
             st.write(f"Error inserting name map: {e}")
             time_sleep(1)
 
-        optimals = init_team_results(players_median, position_requirements, salary_cap, max_team_skaters, teams_to_optimize, 1000)
-        formatted_optimals = format_optimals(optimals, players_median, required_positions)
+        optimals = init_team_results(players_median, position_requirements, salary_cap, max_team_skaters, teams_to_optimize, 1000, players_team_map)
+        formatted_optimals = format_optimals(optimals, players_median, required_positions, players_team_map)
 
         for col in required_positions:
             if col in formatted_optimals.columns: