wzkariampuzha commited on
Commit
d318f36
1 Parent(s): 6e2fbab

Update epi_pipeline.py

Browse files
Files changed (1) hide show
  1. epi_pipeline.py +1 -2
epi_pipeline.py CHANGED
@@ -873,9 +873,8 @@ def streamlit_extraction(search_term:Union[int,str], maxResults:int, filtering:s
873
  sankey_data = (found, relevant, epidemiologic)
874
  #Export the name and GARD ID to the ap for better integration on page.
875
  name = search_term_list[-1].capitalize()
876
- name_gardID = (name, GARD_Search.GARD_dict[search_term_list[-1]])
877
 
878
- return results.sort_values('PROB_OF_EPI', ascending=False), sankey_data, name_gardID
879
 
880
  #Identical to search_term_extraction, except it returns a JSON object instead of a df
881
  def API_extraction(search_term:Union[int,str], maxResults:int, filtering:str, #for abstract search
 
873
  sankey_data = (found, relevant, epidemiologic)
874
  #Export the name and GARD ID to the ap for better integration on page.
875
  name = search_term_list[-1].capitalize()
 
876
 
877
+ return results.sort_values('PROB_OF_EPI', ascending=False), sankey_data, name
878
 
879
  #Identical to search_term_extraction, except it returns a JSON object instead of a df
880
  def API_extraction(search_term:Union[int,str], maxResults:int, filtering:str, #for abstract search