Update resistance_function.py

fmalmeida · web-flow · commit 705250f2e0b0 · 2023-11-03T13:53:34.000-04:00
diff --git a/falmeida_py/resistance_function.py b/falmeida_py/resistance_function.py
@@ -130,10 +130,13 @@ def resistance_stats(bacannot_summary):
                     #
                     # check for rgi orthologies
                     #
-                    if gene in bacannot_summary[sample]['resistance']['rgi']:
-                        bacannot_summary[sample]['resistance']['amrfinderplus'][gene]['card_aro'] = bacannot_summary[sample]['resistance']['rgi'][gene]['card_aro']
-                    else:
-                        bacannot_summary[sample]['resistance']['amrfinderplus'][gene]['card_aro'] = None
+                    try:
+                      if gene in bacannot_summary[sample]['resistance']['rgi']:
+                          bacannot_summary[sample]['resistance']['amrfinderplus'][gene]['card_aro'] = bacannot_summary[sample]['resistance']['rgi'][gene]['card_aro']
+                      else:
+                          bacannot_summary[sample]['resistance']['amrfinderplus'][gene]['card_aro'] = None
+                    except:
+                      bacannot_summary[sample]['resistance']['amrfinderplus'][gene]['card_aro'] = None
             
             #################
             ### resfinder ###
