chore: add environment configs

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

stoix/configs/env/brax/ant.yaml

@@ -0,0 +1,10 @@
+# ---Environment Configs---
+env_name: brax
+
+scenario:
+  name: ant
+  task_name: ant
+
+kwargs: {
+  backend: spring
+}

stoix/configs/env/brax/halfcheetah.yaml

@@ -0,0 +1,10 @@
+# ---Environment Configs---
+env_name: brax
+
+scenario:
+  name : halfcheetah
+  task_name: halfcheetah
+
+kwargs: {
+  backend: spring
+}

stoix/configs/env/brax/humanoid.yaml

@@ -0,0 +1,10 @@
+# ---Environment Configs---
+env_name: brax
+
+scenario:
+  name: humanoid
+  task_name: humanoid
+
+kwargs: {
+  backend: spring
+}

stoix/configs/env/gymnax/asterix.yaml

@@ -0,0 +1,8 @@
+# ---Environment Configs---
+env_name: gymnax
+
+scenario:
+  name: Asterix-MinAtar
+  task_name: asterix
+
+kwargs: {}

stoix/configs/env/gymnax/breakout.yaml

@@ -0,0 +1,8 @@
+# ---Environment Configs---
+env_name: gymnax
+
+scenario:
+  name: Breakout-MinAtar
+  task_name: breakout
+
+kwargs: {}

stoix/configs/env/gymnax/cartpole.yaml

@@ -0,0 +1,8 @@
+# ---Environment Configs---
+env_name: gymnax
+
+scenario:
+  name: CartPole-v1
+  task_name: cartpole
+
+kwargs: {}

stoix/configs/env/gymnax/freeway.yaml

@@ -0,0 +1,8 @@
+# ---Environment Configs---
+env_name: gymnax
+
+scenario:
+  name: Freeway-MinAtar
+  task_name: freeway
+
+kwargs: {}

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: {}

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: {}

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

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

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.

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: {}

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

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

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: {}

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: {}

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: {}

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: {}