From 110c2b94756a5d97aea030d4a1d3ae351a3d8288 Mon Sep 17 00:00:00 2001 From: EdanToledo Date: Sat, 17 Feb 2024 11:30:20 +0000 Subject: [PATCH] chore: add environment configs --- requirements/requirements-dev.txt | 2 +- stoix/configs/env/brax/ant.yaml | 10 ++++++++++ stoix/configs/env/brax/halfcheetah.yaml | 10 ++++++++++ stoix/configs/env/brax/humanoid.yaml | 10 ++++++++++ stoix/configs/env/gymnax/asterix.yaml | 8 ++++++++ stoix/configs/env/gymnax/breakout.yaml | 8 ++++++++ stoix/configs/env/gymnax/cartpole.yaml | 8 ++++++++ stoix/configs/env/gymnax/freeway.yaml | 8 ++++++++ stoix/configs/env/gymnax/mountain_car.yaml | 8 ++++++++ stoix/configs/env/gymnax/space_invaders.yaml | 8 ++++++++ stoix/configs/env/jaxmarl/mabrax.yaml | 12 ++++++++++++ stoix/configs/env/jaxmarl/mpe.yaml | 9 +++++++++ stoix/configs/env/jaxmarl/smax.yaml | 12 ++++++++++++ stoix/configs/env/jumanji/2048.yaml | 11 +++++++++++ stoix/configs/env/jumanji/connector.yaml | 15 +++++++++++++++ stoix/configs/env/jumanji/rware.yaml | 19 +++++++++++++++++++ stoix/configs/env/jumanji/snake.yaml | 11 +++++++++++ stoix/configs/env/jumanji/sokoban.yaml | 11 +++++++++++ .../env/minigrid/minigrid_doorkey_5x5.yaml | 7 +++++++ .../env/minigrid/minigrid_empty_6x6.yaml | 7 +++++++ 20 files changed, 193 insertions(+), 1 deletion(-) create mode 100644 stoix/configs/env/brax/ant.yaml create mode 100644 stoix/configs/env/brax/halfcheetah.yaml create mode 100644 stoix/configs/env/brax/humanoid.yaml create mode 100644 stoix/configs/env/gymnax/asterix.yaml create mode 100644 stoix/configs/env/gymnax/breakout.yaml create mode 100644 stoix/configs/env/gymnax/cartpole.yaml create mode 100644 stoix/configs/env/gymnax/freeway.yaml create mode 100644 stoix/configs/env/gymnax/mountain_car.yaml create mode 100644 stoix/configs/env/gymnax/space_invaders.yaml create mode 100644 stoix/configs/env/jaxmarl/mabrax.yaml create mode 100644 stoix/configs/env/jaxmarl/mpe.yaml create mode 100644 stoix/configs/env/jaxmarl/smax.yaml create mode 100644 stoix/configs/env/jumanji/2048.yaml create mode 100644 stoix/configs/env/jumanji/connector.yaml create mode 100644 stoix/configs/env/jumanji/rware.yaml create mode 100644 stoix/configs/env/jumanji/snake.yaml create mode 100644 stoix/configs/env/jumanji/sokoban.yaml create mode 100644 stoix/configs/env/minigrid/minigrid_doorkey_5x5.yaml create mode 100644 stoix/configs/env/minigrid/minigrid_empty_6x6.yaml diff --git a/requirements/requirements-dev.txt b/requirements/requirements-dev.txt index 8dddf679..b8240ecf 100644 --- a/requirements/requirements-dev.txt +++ b/requirements/requirements-dev.txt @@ -7,7 +7,7 @@ livereload mkdocs==1.2.3 mkdocs-git-revision-date-plugin mkdocs-include-markdown-plugin -mkdocs-material +mkdocs-material==8.2.7 mkdocs-mermaid2-plugin==0.6.0 mkdocstrings==0.18.0 mknotebooks==0.7.1 diff --git a/stoix/configs/env/brax/ant.yaml b/stoix/configs/env/brax/ant.yaml new file mode 100644 index 00000000..13b299f2 --- /dev/null +++ b/stoix/configs/env/brax/ant.yaml @@ -0,0 +1,10 @@ +# ---Environment Configs--- +env_name: brax + +scenario: + name: ant + task_name: ant + +kwargs: { + backend: spring +} diff --git a/stoix/configs/env/brax/halfcheetah.yaml b/stoix/configs/env/brax/halfcheetah.yaml new file mode 100644 index 00000000..c3c6b3b9 --- /dev/null +++ b/stoix/configs/env/brax/halfcheetah.yaml @@ -0,0 +1,10 @@ +# ---Environment Configs--- +env_name: brax + +scenario: + name : halfcheetah + task_name: halfcheetah + +kwargs: { + backend: spring +} diff --git a/stoix/configs/env/brax/humanoid.yaml b/stoix/configs/env/brax/humanoid.yaml new file mode 100644 index 00000000..8628dd06 --- /dev/null +++ b/stoix/configs/env/brax/humanoid.yaml @@ -0,0 +1,10 @@ +# ---Environment Configs--- +env_name: brax + +scenario: + name: humanoid + task_name: humanoid + +kwargs: { + backend: spring +} diff --git a/stoix/configs/env/gymnax/asterix.yaml b/stoix/configs/env/gymnax/asterix.yaml new file mode 100644 index 00000000..3b81a8a7 --- /dev/null +++ b/stoix/configs/env/gymnax/asterix.yaml @@ -0,0 +1,8 @@ +# ---Environment Configs--- +env_name: gymnax + +scenario: + name: Asterix-MinAtar + task_name: asterix + +kwargs: {} diff --git a/stoix/configs/env/gymnax/breakout.yaml b/stoix/configs/env/gymnax/breakout.yaml new file mode 100644 index 00000000..d6e3f50d --- /dev/null +++ b/stoix/configs/env/gymnax/breakout.yaml @@ -0,0 +1,8 @@ +# ---Environment Configs--- +env_name: gymnax + +scenario: + name: Breakout-MinAtar + task_name: breakout + +kwargs: {} diff --git a/stoix/configs/env/gymnax/cartpole.yaml b/stoix/configs/env/gymnax/cartpole.yaml new file mode 100644 index 00000000..554d5835 --- /dev/null +++ b/stoix/configs/env/gymnax/cartpole.yaml @@ -0,0 +1,8 @@ +# ---Environment Configs--- +env_name: gymnax + +scenario: + name: CartPole-v1 + task_name: cartpole + +kwargs: {} diff --git a/stoix/configs/env/gymnax/freeway.yaml b/stoix/configs/env/gymnax/freeway.yaml new file mode 100644 index 00000000..08ccc873 --- /dev/null +++ b/stoix/configs/env/gymnax/freeway.yaml @@ -0,0 +1,8 @@ +# ---Environment Configs--- +env_name: gymnax + +scenario: + name: Freeway-MinAtar + task_name: freeway + +kwargs: {} diff --git a/stoix/configs/env/gymnax/mountain_car.yaml b/stoix/configs/env/gymnax/mountain_car.yaml new file mode 100644 index 00000000..fafbef5b --- /dev/null +++ b/stoix/configs/env/gymnax/mountain_car.yaml @@ -0,0 +1,8 @@ +# ---Environment Configs--- +env_name: gymnax + +scenario: + name: MountainCar-v0 + task_name: mountain_car + +kwargs: {} diff --git a/stoix/configs/env/gymnax/space_invaders.yaml b/stoix/configs/env/gymnax/space_invaders.yaml new file mode 100644 index 00000000..81822830 --- /dev/null +++ b/stoix/configs/env/gymnax/space_invaders.yaml @@ -0,0 +1,8 @@ +# ---Environment Configs--- +env_name: gymnax + +scenario: + name: SpaceInvaders-MinAtar + task_name: space_invaders + +kwargs: {} diff --git a/stoix/configs/env/jaxmarl/mabrax.yaml b/stoix/configs/env/jaxmarl/mabrax.yaml new file mode 100644 index 00000000..2b23dce5 --- /dev/null +++ b/stoix/configs/env/jaxmarl/mabrax.yaml @@ -0,0 +1,12 @@ +# ---Environment Configs--- +env_name: mabrax + +scenario: + name: ant_4x2 + task_name: ant_4x2 + +kwargs: + episode_length : 1000 + action_repeat: 1 + auto_reset: False + backend: spring diff --git a/stoix/configs/env/jaxmarl/mpe.yaml b/stoix/configs/env/jaxmarl/mpe.yaml new file mode 100644 index 00000000..89270e7b --- /dev/null +++ b/stoix/configs/env/jaxmarl/mpe.yaml @@ -0,0 +1,9 @@ +# ---Environment Configs--- +env_name: mpe + +scenario: + name: MPE_simple_v3 + task_name: mpe_simple + +kwargs: + num_agents : 2 diff --git a/stoix/configs/env/jaxmarl/smax.yaml b/stoix/configs/env/jaxmarl/smax.yaml new file mode 100644 index 00000000..427640b4 --- /dev/null +++ b/stoix/configs/env/jaxmarl/smax.yaml @@ -0,0 +1,12 @@ +# ---Environment Configs--- +env_name: smax +# Possible scenarios: +# 2s3z | 3s5z | 5m_vs_6m | 10m_vs_11m | 27m_vs_30m | 3s5z_vs_3s6z | 3s_vs_5z | 6h_vs_8z | smacv2_5_units | smacv2_10_units | smacv2_20_units +scenario: + name: HeuristicEnemySMAX + task_name: 2s3z + +kwargs: + see_enemy_actions: True # Whether to enable enemy vision. If True, the enemy will be able to see the actions of the agent. + walls_cause_death: True # Whether to kill the agent if it collides with a wall. + attack_mode: "closest" # "closest" or "random". Whether to attack the closest enemy or a random enemy. diff --git a/stoix/configs/env/jumanji/2048.yaml b/stoix/configs/env/jumanji/2048.yaml new file mode 100644 index 00000000..5f165b69 --- /dev/null +++ b/stoix/configs/env/jumanji/2048.yaml @@ -0,0 +1,11 @@ +# ---Environment Configs--- +env_name: jumanji +observation_attribute : board +flatten_observation: True +multi_agent : False + +scenario: + name: Game2048-v1 + task_name: game_2048 + +kwargs: {} diff --git a/stoix/configs/env/jumanji/connector.yaml b/stoix/configs/env/jumanji/connector.yaml new file mode 100644 index 00000000..8353c4a3 --- /dev/null +++ b/stoix/configs/env/jumanji/connector.yaml @@ -0,0 +1,15 @@ +# ---Environment Configs--- +env_name: jumanji +observation_attribute : grid +flatten_observation: True +multi_agent : True + +scenario: + name: Connector-v2 + task_name: connector + +kwargs: + generator: + _target_: jumanji.environments.routing.connector.generator.RandomWalkGenerator + grid_size: 6 + num_agents: 2 diff --git a/stoix/configs/env/jumanji/rware.yaml b/stoix/configs/env/jumanji/rware.yaml new file mode 100644 index 00000000..47462e65 --- /dev/null +++ b/stoix/configs/env/jumanji/rware.yaml @@ -0,0 +1,19 @@ +# ---Environment Configs--- +env_name: jumanji +observation_attribute : agents_view +flatten_observation: True +multi_agent : True + +scenario: + name: RobotWarehouse-v0 + task_name: robot_warehouse + +kwargs: + generator: + _target_: jumanji.environments.routing.robot_warehouse.generator.RandomGenerator + column_height: 8 + shelf_rows: 2 + shelf_columns: 3 + num_agents: 4 + sensor_range: 1 + request_queue_size: 8 diff --git a/stoix/configs/env/jumanji/snake.yaml b/stoix/configs/env/jumanji/snake.yaml new file mode 100644 index 00000000..9fea0a02 --- /dev/null +++ b/stoix/configs/env/jumanji/snake.yaml @@ -0,0 +1,11 @@ +# ---Environment Configs--- +env_name: jumanji +observation_attribute : grid +flatten_observation: True +multi_agent : False + +scenario: + name: Snake-v1 + task_name: snake + +kwargs: {} diff --git a/stoix/configs/env/jumanji/sokoban.yaml b/stoix/configs/env/jumanji/sokoban.yaml new file mode 100644 index 00000000..fb5a4422 --- /dev/null +++ b/stoix/configs/env/jumanji/sokoban.yaml @@ -0,0 +1,11 @@ +# ---Environment Configs--- +env_name: jumanji +observation_attribute : grid +flatten_observation: True +multi_agent : False + +scenario: + name: Sokoban-v0 + task_name: sokoban + +kwargs: {} diff --git a/stoix/configs/env/minigrid/minigrid_doorkey_5x5.yaml b/stoix/configs/env/minigrid/minigrid_doorkey_5x5.yaml new file mode 100644 index 00000000..987784da --- /dev/null +++ b/stoix/configs/env/minigrid/minigrid_doorkey_5x5.yaml @@ -0,0 +1,7 @@ +# ---Environment Configs--- +env_name: MiniGrid-DoorKey-5x5 + +scenario: + task_name: minigrid_doorkey_5x5 + +kwargs: {} diff --git a/stoix/configs/env/minigrid/minigrid_empty_6x6.yaml b/stoix/configs/env/minigrid/minigrid_empty_6x6.yaml new file mode 100644 index 00000000..2dea98a7 --- /dev/null +++ b/stoix/configs/env/minigrid/minigrid_empty_6x6.yaml @@ -0,0 +1,7 @@ +# ---Environment Configs--- +env_name: MiniGrid-Empty-6x6 + +scenario: + task_name: minigrid_empty_6x6 + +kwargs: {}