tasks with dead agents eliminated

facebookresearch · Oct 29, 2024 · b70002b · b70002b
1 parent 41afc3f
commit b70002b
Show file tree

Hide file tree

Showing 11 changed files with 6 additions and 155 deletions.
diff --git a/benchmarl/conf/task/magent/battle.yaml b/benchmarl/conf/task/magent/battle.yaml
diff --git a/benchmarl/conf/task/magent/battlefield.yaml b/benchmarl/conf/task/magent/battlefield.yaml
diff --git a/benchmarl/conf/task/magent/combined_arms.yaml b/benchmarl/conf/task/magent/combined_arms.yaml
diff --git a/benchmarl/conf/task/magent/gather.yaml b/benchmarl/conf/task/magent/gather.yaml
diff --git a/benchmarl/conf/task/magent/tiger_deer.yaml b/benchmarl/conf/task/magent/tiger_deer.yaml
diff --git a/benchmarl/environments/magent/battle.py b/benchmarl/environments/magent/battle.py
diff --git a/benchmarl/environments/magent/battlefield.py b/benchmarl/environments/magent/battlefield.py
diff --git a/benchmarl/environments/magent/combined_arms.py b/benchmarl/environments/magent/combined_arms.py
diff --git a/benchmarl/environments/magent/common.py b/benchmarl/environments/magent/common.py
@@ -18,11 +18,11 @@ class MAgentTask(Task):
     """Enum for MAgent2 tasks."""
 
     ADVERSARIAL_PURSUIT = None
-    BATTLE = None
-    BATTLEFIELD = None
-    COMBINED_ARMS = None
-    GATHER = None
-    TIGER_DEER = None
+    # BATTLE = None
+    # BATTLEFIELD = None
+    # COMBINED_ARMS = None
+    # GATHER = None
+    # TIGER_DEER = None
 
     def get_env_fun(
             self,
@@ -37,7 +37,7 @@ def get_env_fun(
             return_state=True,
             seed=seed,
             done_on_any=False,
-            use_mask=True,
+            use_mask=False,
             device=device
         )
 

diff --git a/benchmarl/environments/magent/gather.py b/benchmarl/environments/magent/gather.py
diff --git a/benchmarl/environments/magent/tiger_deer.py b/benchmarl/environments/magent/tiger_deer.py