unset don on any

facebookresearch · Oct 9, 2024 · 8d397ff · 8d397ff
1 parent 76d911d
commit 8d397ff
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/benchmarl/environments/magent/common.py b/benchmarl/environments/magent/common.py
@@ -45,7 +45,8 @@ def get_env_fun(
             env=self.__get_env(),
             return_state=True,
             seed=seed,
-            done_on_any=True,
+            done_on_any=False,
+            use_mask=True,
             device=device
         )