File tree Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Expand file tree Collapse file tree 1 file changed +6
-6
lines changed Original file line number Diff line number Diff line change 33from matplotlib .figure import Figure
44from mesa .experimental import JupyterViz
55from sugarscape_g1mt .model import SugarscapeG1mt
6- from sugarscape_g1mt .resource_agents import Sugar
76from sugarscape_g1mt .trader_agents import Trader
87
98
@@ -22,11 +21,12 @@ def portray(g):
2221 layers ["trader" ]["y" ].append (j )
2322 else :
2423 # Don't visualize resource with value <= 1.
25- value = agent .amount if agent .amount > 1 else np .nan
26- if isinstance (agent , Sugar ):
27- layers ["sugar" ][i ][j ] = value
28- else :
29- layers ["spice" ][i ][j ] = value
24+ layers ["sugar" ][i ][j ] = (
25+ agent .sugar_amount if agent .sugar_amount > 1 else np .nan
26+ )
27+ layers ["spice" ][i ][j ] = (
28+ agent .spice_amount if agent .spice_amount > 1 else np .nan
29+ )
3030 return layers
3131
3232 fig = Figure ()
You can’t perform that action at this time.
0 commit comments