VyLala commited on
Commit
687a800
·
verified ·
1 Parent(s): ba26203

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +4 -4
pipeline.py CHANGED
@@ -910,10 +910,10 @@ async def pipeline_with_gemini(accessions,stop_flag=None, save_df=None, niche_ca
910
  acc_score["country"][country.lower()] = [method_used + country_explanation]
911
  # predicted country is non unknown
912
  acc_score["signals"]["predicted_country"] = country.lower()
913
- else:
914
- # predicted country is unknown
915
- acc_score["signals"]["predicted_country"] = "unknown"
916
- acc_score["signals"]["known_failure_pattern"] = True
917
  # for sample type
918
  elif pred_out == "modern/ancient/unknown":
919
  sample_type = predicted_outputs[pred_out]["answer"]
 
910
  acc_score["country"][country.lower()] = [method_used + country_explanation]
911
  # predicted country is non unknown
912
  acc_score["signals"]["predicted_country"] = country.lower()
913
+ else:
914
+ # predicted country is unknown
915
+ acc_score["signals"]["predicted_country"] = "unknown"
916
+ acc_score["signals"]["known_failure_pattern"] = True
917
  # for sample type
918
  elif pred_out == "modern/ancient/unknown":
919
  sample_type = predicted_outputs[pred_out]["answer"]