mtyrrell commited on
Commit
a49bcd0
·
1 Parent(s): 2906ce0

More fixes to pipeline logic

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. modules/pipeline.py +2 -2
app.py CHANGED
@@ -130,6 +130,7 @@ def main():
130
  horizontal = False)
131
 
132
  sens_level = sens_options[sens_input]
 
133
 
134
  with st.expander("ℹ️ - About this app", expanded=False):
135
  st.write(
 
130
  horizontal = False)
131
 
132
  sens_level = sens_options[sens_input]
133
+ logger.info(f"Sensitivity level applied: {sens_level}")
134
 
135
  with st.expander("ℹ️ - About this app", expanded=False):
136
  st.write(
modules/pipeline.py CHANGED
@@ -318,11 +318,11 @@ def process_data(uploaded_file, sens_level, azure_client, azure_deployment):
318
  not any(sector in [x['SECTOR1'], x['SECTOR2']] for sector in sector_classes) or
319
  x['word_length_check'] == True or
320
  x['duplicate_check'] == True or
321
- x['pred_score'] <= sens_level )
322
  else 'PRE-ASSESSMENT' if sens_level <= x['pred_score'] <= sens_level+1
323
  else 'FULL-ASSESSMENT' if x['pred_score'] > sens_level+2
324
  else 'ERROR', axis=1)
325
- logger.info(f"Sensitivity level applied: {sens_level}")
326
  # Reorder columns in final dataframe
327
  column_order = ['id', 'organization', 'org_renamed', 'concept_count', 'duplicate_check', 'scope_txt', 'tech_txt', 'fin_txt', 'maf_funding', 'cont_public',
328
  'cont_private', 'cont_other', 'scope_lab1', 'scope_lab2', 'tech_lab1',
 
318
  not any(sector in [x['SECTOR1'], x['SECTOR2']] for sector in sector_classes) or
319
  x['word_length_check'] == True or
320
  x['duplicate_check'] == True or
321
+ x['pred_score'] < sens_level)
322
  else 'PRE-ASSESSMENT' if sens_level <= x['pred_score'] <= sens_level+1
323
  else 'FULL-ASSESSMENT' if x['pred_score'] > sens_level+2
324
  else 'ERROR', axis=1)
325
+
326
  # Reorder columns in final dataframe
327
  column_order = ['id', 'organization', 'org_renamed', 'concept_count', 'duplicate_check', 'scope_txt', 'tech_txt', 'fin_txt', 'maf_funding', 'cont_public',
328
  'cont_private', 'cont_other', 'scope_lab1', 'scope_lab2', 'tech_lab1',