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

Refactor portfolio storage in app.py to filter by salary cap before saving to compressed format. This change optimizes memory usage and ensures only eligible entries are retained.

Browse files
Files changed (1) hide show
  1. app.py +1 -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['origin_portfolio'] = st.session_state['portfolio']
781
  buffer = io.BytesIO()
782
  st.session_state['portfolio'].to_parquet(buffer, compression='snappy')
783
  st.session_state['origin_portfolio'] = buffer.getvalue()
 
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()