add try catch

CanDIG · Mar 8, 2024 · d2d5a03 · d2d5a03
1 parent cd22b9b
commit d2d5a03
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/query_server/query_operations.py b/query_server/query_operations.py
@@ -95,8 +95,12 @@ def get_summary_stats(donors, headers):
     treatment_type_count = {}
     for treatment in treatments:
         if treatment["submitter_donor_id"] in donors_by_id:
-            for treatment_type in treatment["treatment_type"]:
-                add_or_increment(treatment_type_count, treatment_type)
+            try:
+                for treatment_type in treatment["treatment_type"]:
+                    add_or_increment(treatment_type_count, treatment_type)
+            except TypeError as e:
+                print(e)
+                pass
 
     return {
         'age_at_diagnosis': age_at_diagnosis,