collusion test

BioData-PT · Nov 29, 2024 · ea785d3 · ea785d3
1 parent aa7a9a9
commit ea785d3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/beacon/scripts/collusion_test.py b/beacon/scripts/collusion_test.py
@@ -99,7 +99,8 @@ def update_user_budget_to_initial(individual_id):
         )
         print("!!!!!!!!!!!!!!!!!!!!!!!!!!!")
         print("THE BUDGET WAS UPDATED, SO 1 MORE USER")
-        print("The budget is now: ", client.beacon.get_collection('budget').find_one({"individualId": individual_id})['budget'])
+        budget_info = client.beacon.get_collection('budget').find_one({"individualId": individual_id})
+        print("The budget is now: ", budget_info['budget'])
         print("!!!!!!!!!!!!!!!!!!!!!!!!!!!")
     except Exception as e:
         return None