chore: fix typo and add config

EdanToledo · Jul 9, 2024 · 4d06683 · 4d06683
1 parent 6505baf
commit 4d06683
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/stoix/configs/env/navix/empty_5x5.yaml b/stoix/configs/env/navix/empty_5x5.yaml
@@ -2,7 +2,7 @@
 env_name: navix
 scenario:
   name: Navix-Empty-5x5-v0
-  task_name: navix-dempty-5x5-v0
+  task_name: navix-empty-5x5-v0
 
 kwargs: {}
 

diff --git a/stoix/configs/env/xland_minigrid/empty_5x5.yaml b/stoix/configs/env/xland_minigrid/empty_5x5.yaml
@@ -0,0 +1,16 @@
+# ---Environment Configs---
+env_name: xland_minigrid
+scenario:
+  name: MiniGrid-Empty-5x5
+  task_name: minigrid_empty_5x5
+
+kwargs: {}
+
+# Defines the metric that will be used to evaluate the performance of the agent.
+# This metric is returned at the end of an experiment and can be used for hyperparameter tuning.
+eval_metric: episode_return
+
+
+# Optional wrapper to flatten the observation space.
+wrapper:
+  _target_: stoix.wrappers.transforms.FlattenObservationWrapper