Fix hex_ant and hex_snowflake: migrate to Mesa 3.x visualization API

examples/hex_ant/app.py

@@ -1,10 +1,9 @@
 import matplotlib.pyplot as plt
+from agent import AntState
 from matplotlib.colors import LinearSegmentedColormap, ListedColormap
 from mesa.visualization import SolaraViz, make_space_component
 from mesa.visualization.components import PropertyLayerStyle
-
-from .agent import AntState
-from .model import AntForaging
+from model import AntForaging
 
 plt.rcParams["figure.figsize"] = (10, 10)
 

examples/hex_ant/model.py

@@ -1,8 +1,7 @@
 import mesa
+from agent import Ant
 from mesa.discrete_space import HexGrid
 
-from .agent import Ant
-
 
 class AntForaging(mesa.Model):
     """

examples/hex_snowflake/hex_snowflake/app.py

@@ -0,0 +1,21 @@
+from hex_snowflake.cell import Cell
+from hex_snowflake.model import HexSnowflake
+from mesa.visualization import SolaraViz, make_space_component
+
+
+def agent_portrayal(agent):
+    if agent is None:
+        return
+
+    if isinstance(agent, Cell):
+        color = "tab:green" if agent.state == Cell.ALIVE else "tab:grey"
+        return {"color": color, "size": 20}
+
+
+model = HexSnowflake()
+
+page = SolaraViz(
+    model,
+    components=[make_space_component(agent_portrayal=agent_portrayal)],
+    name="Hex Snowflake",
+)

examples/hex_snowflake/run.py

@@ -1,3 +0,0 @@
-from hex_snowflake.server import server
-
-server.launch(open_browser=True)