James McCool commited on
Commit
8ee5fa8
·
1 Parent(s): 04606ac

Enhance data export functionality by initializing and filtering 'working_seed' based on team and stack variables in session state

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -241,6 +241,13 @@ with tab2:
241
 
242
 
243
  if st.button("Prepare data export", key='data_export'):
 
 
 
 
 
 
 
244
  data_export = st.session_state.working_seed.copy()
245
  for col in data_export[:, 0:9]:
246
  data_export[col] = data_export[col].map(dk_id_dict)
 
241
 
242
 
243
  if st.button("Prepare data export", key='data_export'):
244
+ if 'working_seed' in st.session_state:
245
+ st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
246
+ st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
247
+ elif 'working_seed' not in st.session_state:
248
+ st.session_state.working_seed = DK_seed.copy()
249
+ st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 11], team_var2)]
250
+ st.session_state.working_seed = st.session_state.working_seed[np.isin(st.session_state.working_seed[:, 12], stack_var2)]
251
  data_export = st.session_state.working_seed.copy()
252
  for col in data_export[:, 0:9]:
253
  data_export[col] = data_export[col].map(dk_id_dict)