fix: Fallback to None when attribute is missing

projectmesa · Jan 24, 2024 · b547ee3 · b547ee3
1 parent 2162451
commit b547ee3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/bank_reserves/bank_reserves/model.py b/examples/bank_reserves/bank_reserves/model.py
@@ -134,7 +134,7 @@ def __init__(
                 "Money": get_total_money,
                 "Loans": get_total_loans,
             },
-            agent_reporters={"Wealth": lambda x: x.wealth},
+            agent_reporters={"Wealth": lambda x: getattr(x, "wealth", None)},
         )
 
         # create a single bank for the model

diff --git a/examples/charts/charts/model.py b/examples/charts/charts/model.py
@@ -117,7 +117,7 @@ def __init__(
                 "Money": get_total_money,
                 "Loans": get_total_loans,
             },
-            agent_reporters={"Wealth": lambda x: x.wealth},
+            agent_reporters={"Wealth": lambda x: getattr(x, "wealth", None)},
         )
 
         # create a single bank for the model