diff --git a/examples/virus_antibody/app.py b/examples/virus_antibody/app.py index 28ce536f..f05588b6 100644 --- a/examples/virus_antibody/app.py +++ b/examples/virus_antibody/app.py @@ -1,13 +1,6 @@ -import os -import sys import weakref -from agents import AntibodyAgent, VirusAgent from matplotlib.markers import MarkerStyle -from model import VirusAntibodyModel - -sys.path.insert(0, os.path.abspath("../../../mesa")) - from mesa.experimental.devs import ABMSimulator from mesa.visualization import ( Slider, @@ -15,6 +8,8 @@ make_plot_component, make_space_component, ) +from virus_antibody.agents import AntibodyAgent, VirusAgent +from virus_antibody.model import VirusAntibodyModel # Style and portrayals MARKER_CACHE = {} diff --git a/examples/virus_antibody/virus_antibody/__init__.py b/examples/virus_antibody/virus_antibody/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/examples/virus_antibody/agents.py b/examples/virus_antibody/virus_antibody/agents.py similarity index 100% rename from examples/virus_antibody/agents.py rename to examples/virus_antibody/virus_antibody/agents.py diff --git a/examples/virus_antibody/model.py b/examples/virus_antibody/virus_antibody/model.py similarity index 100% rename from examples/virus_antibody/model.py rename to examples/virus_antibody/virus_antibody/model.py