James McCool commited on
Commit
54ce19d
·
1 Parent(s): 1b89d01

Refactor exposure_spread function to correctly apply team removal logic to comparable players, ensuring accurate filtering in player comparisons.

Browse files
Files changed (1) hide show
  1. global_func/exposure_spread.py +3 -3
global_func/exposure_spread.py CHANGED
@@ -245,11 +245,11 @@ def exposure_spread(working_frame, exposure_player, exposure_target, exposure_st
245
  comparable_players = comparable_players[comparable_players['player_names'] != exposure_player]
246
 
247
  if remove_teams is not None:
248
- remove_mask = player_rows.apply(
249
  lambda row: not any(team in list(row) for team in remove_teams), axis=1
250
  )
251
- player_rows = player_rows[remove_mask]
252
-
253
  # Create a list of comparable players
254
  comparable_player_list = comparable_players['player_names'].tolist()
255
 
 
245
  comparable_players = comparable_players[comparable_players['player_names'] != exposure_player]
246
 
247
  if remove_teams is not None:
248
+ remove_mask = comparable_players.apply(
249
  lambda row: not any(team in list(row) for team in remove_teams), axis=1
250
  )
251
+ comparable_players = comparable_players[remove_mask]
252
+
253
  # Create a list of comparable players
254
  comparable_player_list = comparable_players['player_names'].tolist()
255