James McCool commited on
Commit
d864275
·
1 Parent(s): e66d221

Update portfolio management in app.py to reintroduce salary cap filtering for working frame. This change ensures that only eligible players are loaded into the working frame, enhancing data integrity and memory efficiency.

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -777,7 +777,7 @@ if selected_tab == 'Data Load':
777
  st.session_state['map_dict'] = create_memory_efficient_mappings(st.session_state['projections_df'], site_var, type_var, sport_var)
778
 
779
  # Store portfolio in compressed format and clean up
780
- st.session_state['portfolio'] = st.session_state['portfolio'][st.session_state['portfolio']['salary'] <= salary_max]
781
  buffer = io.BytesIO()
782
  st.session_state['portfolio'].to_parquet(buffer, compression='snappy')
783
  st.session_state['origin_portfolio'] = buffer.getvalue()
@@ -1327,6 +1327,7 @@ if selected_tab == 'Manage Portfolio':
1327
  if 'working_frame' not in st.session_state:
1328
  st.session_state['settings_base'] = True
1329
  st.session_state['working_frame'] = pd.read_parquet(io.BytesIO(st.session_state['origin_portfolio']))
 
1330
  st.session_state['player_columns'] = [col for col in st.session_state['working_frame'].columns if col not in excluded_cols]
1331
 
1332
  # Use vectorized calculation function
 
777
  st.session_state['map_dict'] = create_memory_efficient_mappings(st.session_state['projections_df'], site_var, type_var, sport_var)
778
 
779
  # Store portfolio in compressed format and clean up
780
+ # st.session_state['portfolio'] = st.session_state['portfolio'][st.session_state['portfolio']['salary'] <= salary_max]
781
  buffer = io.BytesIO()
782
  st.session_state['portfolio'].to_parquet(buffer, compression='snappy')
783
  st.session_state['origin_portfolio'] = buffer.getvalue()
 
1327
  if 'working_frame' not in st.session_state:
1328
  st.session_state['settings_base'] = True
1329
  st.session_state['working_frame'] = pd.read_parquet(io.BytesIO(st.session_state['origin_portfolio']))
1330
+ st.session_state['working_frame'] = st.session_state['working_frame'][st.session_state['working_frame']['salary'] <= salary_max]
1331
  st.session_state['player_columns'] = [col for col in st.session_state['working_frame'].columns if col not in excluded_cols]
1332
 
1333
  # Use vectorized calculation function