make all basic examples importable directly by name

Author: Corvince

examples/__init__.py

@@ -0,0 +1,13 @@
+from .boid_flockers.model import BoidFlockers
+from .boltzmann_wealth_model.model import BoltzmannWealthModel
+from .conways_game_of_life.model import ConwaysGameOfLife
+from .schelling.model import Schelling
+from .virus_on_network.model import VirusOnNetwork
+
+__all__ = [
+    "BoidFlockers",
+    "BoltzmannWealthModel",
+    "ConwaysGameOfLife",
+    "Schelling",
+    "VirusOnNetwork",
+]

examples/basic/__init__.py

@@ -1,7 +1,7 @@
-from model import BoidFlockers
-
 from mesa.visualization import Slider, SolaraViz, make_space_matplotlib
 
+from .model import BoidFlockers
+
 
 def boid_draw(agent):
     if not agent.neighbors:  # Only for the first Frame

examples/basic/boid_flockers/app.py

@@ -5,10 +5,11 @@
 """
 
 import numpy as np
-from agents import Boid
 
 import mesa
 
+from .agents import Boid
+
 
 class BoidFlockers(mesa.Model):
     """Flocker model class. Handles agent creation, placement and scheduling."""

examples/basic/boid_flockers/model.py

@@ -1,11 +1,11 @@
-from model import BoltzmannWealthModel
-
 from mesa.visualization import (
     SolaraViz,
     make_plot_measure,
     make_space_matplotlib,
 )
 
+from .model import BoltzmannWealthModel
+
 
 def agent_portrayal(agent):
     size = 10

examples/basic/boltzmann_wealth_model/app.py

@@ -1,7 +1,7 @@
-from agents import MoneyAgent
-
 import mesa
 
+from .agents import MoneyAgent
+
 
 class BoltzmannWealthModel(mesa.Model):
     """A simple model of an economy where agents exchange currency at random.

examples/basic/boltzmann_wealth_model/model.py

@@ -1,8 +1,8 @@
-from agents import Cell
-
 from mesa import Model
 from mesa.space import SingleGrid
 
+from .agents import Cell
+
 
 class ConwaysGameOfLife(Model):
     """Represents the 2-dimensional array of cells in Conway's Game of Life."""

examples/basic/conways_game_of_life/model.py

@@ -1,8 +1,8 @@
-from model import ConwaysGameOfLife
-from portrayal import portrayCell
-
 import mesa
 
+from .model import ConwaysGameOfLife
+from .portrayal import portrayCell
+
 # Make a world that is 50x50, on a 250x250 display.
 canvas_element = mesa.visualization.CanvasGrid(portrayCell, 50, 50, 250, 250)
 

examples/basic/conways_game_of_life/server.py

@@ -1,5 +1,4 @@
 import solara
-from model import Schelling
 
 from mesa.visualization import (
     Slider,
@@ -8,6 +7,8 @@
     make_space_matplotlib,
 )
 
+from .model import Schelling
+
 
 def get_happy_agents(model):
     """Display a text count of how many happy agents there are."""

examples/basic/schelling/app.py

@@ -1,8 +1,8 @@
-from agents import SchellingAgent
-
 import mesa
 from mesa import Model
 
+from .agents import SchellingAgent
+
 
 class Schelling(Model):
     """Model class for the Schelling segregation model."""