James McCool commited on
Commit
d94819e
·
1 Parent(s): e0bd286

Refactor portfolio filtering in app.py to convert entries to string and remove invalid values before saving to compressed format. This update enhances data integrity and ensures only valid entries are retained.

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -777,7 +777,8 @@ 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()
 
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'].astype(str)
781
+ st.session_state['portfolio'] = st.session_state['portfolio'][~st.session_state['portfolio'].isin(['', 'nan', 'None', 'NaN']).any(axis=1)].reset_index(drop=True)
782
  buffer = io.BytesIO()
783
  st.session_state['portfolio'].to_parquet(buffer, compression='snappy')
784
  st.session_state['origin_portfolio'] = buffer.getvalue()