from vivarium.engine import InteractiveContext
from vivarium.engine.examples.boids import Population, Movement, Neighbors, Separation, Cohesion, Alignment, plot_boids

sim = InteractiveContext(
   components=[Population(), Movement(), Neighbors(), Separation(), Cohesion(), Alignment()],
   configuration={'population': {'population_size': 500}},
   logging_verbosity=0,
)
sim.take_steps(100)
plot_boids(sim, plot_velocity=True)