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
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 |
-
|
| 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()
|