Skip to content

Commit

Permalink
Bump version
Browse files Browse the repository at this point in the history
  • Loading branch information
michaelkoelle committed Jan 26, 2024
1 parent ef48060 commit 234f7d6
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 1 deletion.
25 changes: 25 additions & 0 deletions example.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
"""Example of how to use the Aquarium environment."""

from marl_aquarium import aquarium_v0

env = aquarium_v0.env(
# draw_force_vectors=True,
# draw_action_vectors=True,
# draw_view_cones=True,
# draw_hit_boxes=True,
# draw_death_circles=True,
)
env.reset(seed=42)

for agent in env.agent_iter():
observation, reward, termination, truncation, info = env.last()

if termination or truncation:
action = None
else:
# this is where you would insert your policy
action = env.action_space(agent).sample()

env.step(action)
env.render()
env.close()
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

setup(
name="marl-aquarium",
version="0.1.8",
version="0.1.9",
packages=find_packages(exclude=["examples"]),
license="MIT",
description="Aquarium: A Comprehensive Framework for Exploring Predator-Prey Dynamics through Multi-Agent Reinforcement Learning Algorithms",
Expand Down

0 comments on commit 234f7d6

Please sign in to comment.