[BugFix] VMAS parameters names (#149)

* fix flocking * fix dispersion
facebookresearch · Nov 27, 2024 · 3d15312 · 3d15312
1 parent 8045155
commit 3d15312
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/benchmarl/conf/task/vmas/dispersion.yaml b/benchmarl/conf/task/vmas/dispersion.yaml
@@ -7,6 +7,6 @@ defaults:
 max_steps: 100
 n_agents: 4
 n_food: 4
-share_rew: True
+share_reward: True
 food_radius: 0.02
 penalise_by_time: False
diff --git a/benchmarl/conf/task/vmas/flocking.yaml b/benchmarl/conf/task/vmas/flocking.yaml
@@ -5,5 +5,5 @@ defaults:
 
 max_steps: 100
 n_agents: 4
-n_targets: 5
+n_obstacles: 5
 collision_reward: -0.1
diff --git a/benchmarl/environments/vmas/dispersion.py b/benchmarl/environments/vmas/dispersion.py
@@ -12,6 +12,6 @@ class TaskConfig:
     max_steps: int = MISSING
     n_agents: int = MISSING
     n_food: int = MISSING
-    share_rew: bool = MISSING
+    share_reward: bool = MISSING
     food_radius: float = MISSING
     penalise_by_time: bool = MISSING
diff --git a/benchmarl/environments/vmas/flocking.py b/benchmarl/environments/vmas/flocking.py
@@ -11,5 +11,5 @@
 class TaskConfig:
     max_steps: int = MISSING
     n_agents: int = MISSING
-    n_targets: int = MISSING
+    n_obstacles: int = MISSING
     collision_reward: float = MISSING