James McCool
commited on
Commit
·
7df001f
1
Parent(s):
40a0be8
Refactor portfolio storage in session state to use compressed format directly
Browse filesUpdated 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.
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['
|
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']
|