James McCool commited on
Commit
7df001f
·
1 Parent(s): 40a0be8

Refactor portfolio storage in session state to use compressed format directly

Browse files

Updated the handling of the portfolio in session state by storing it in a compressed format, improving memory efficiency. Removed the redundant storage of the original portfolio to streamline session state management.

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -591,10 +591,10 @@ if selected_tab == 'Data Load':
591
  st.session_state['map_dict'] = create_memory_efficient_mappings(st.session_state['projections_df'], site_var, type_var, sport_var)
592
 
593
  # Store portfolio in compressed format and clean up
594
- st.session_state['origin_portfolio'] = st.session_state['portfolio']
595
  buffer = io.BytesIO()
596
  st.session_state['portfolio'].to_parquet(buffer, compression='snappy')
597
- st.session_state['origin_portfolio_compressed'] = buffer.getvalue()
598
 
599
  # Clear large objects from session state to free memory
600
  del st.session_state['portfolio'], st.session_state['export_portfolio']
 
591
  st.session_state['map_dict'] = create_memory_efficient_mappings(st.session_state['projections_df'], site_var, type_var, sport_var)
592
 
593
  # Store portfolio in compressed format and clean up
594
+ # st.session_state['origin_portfolio'] = st.session_state['portfolio']
595
  buffer = io.BytesIO()
596
  st.session_state['portfolio'].to_parquet(buffer, compression='snappy')
597
+ st.session_state['origin_portfolio'] = buffer.getvalue()
598
 
599
  # Clear large objects from session state to free memory
600
  del st.session_state['portfolio'], st.session_state['export_portfolio']