fix: Fallback to None when attribute is missing

rht · rht · commit b547ee3381a2 · 2024-01-24T05:50:42.000-05:00
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

Original file line number	Diff line number	Diff line change
`@@ -134,7 +134,7 @@ def __init__(`
`134`	`134`	`"Money": get_total_money,`
`135`	`135`	`"Loans": get_total_loans,`
`136`	`136`	`},`
`137`		`- agent_reporters={"Wealth": lambda x: x.wealth},`
	`137`	`+ agent_reporters={"Wealth": lambda x: getattr(x, "wealth", None)},`
`138`	`138`	`)`
`139`	`139`
`140`	`140`	`# create a single bank for the model`
Original file line number	Diff line number	Diff line change
`@@ -117,7 +117,7 @@ def __init__(`
`117`	`117`	`"Money": get_total_money,`
`118`	`118`	`"Loans": get_total_loans,`
`119`	`119`	`},`
`120`		`- agent_reporters={"Wealth": lambda x: x.wealth},`
	`120`	`+ agent_reporters={"Wealth": lambda x: getattr(x, "wealth", None)},`
`121`	`121`	`)`
`122`	`122`
`123`	`123`	`# create a single bank for the model`