all tests pass AAAAA

Farama-Foundation · Oct 24, 2023 · 873e5c0 · 873e5c0
1 parent 53d2e39
commit 873e5c0
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/momadm_benchmarks/envs/crazyrl/catch/catch.py b/momadm_benchmarks/envs/crazyrl/catch/catch.py
@@ -254,6 +254,8 @@ def _compute_terminated(self):
         for agent in self.agents:
             terminated[agent] = False
 
+        # print('beginning:', type(terminated["agent_0"]))
+
         for agent in self.agents:
             # collision between two drones
             for other_agent in self.agents:
@@ -275,6 +277,8 @@ def _compute_terminated(self):
                     terminated[other_agent] = True
                 self.agents = []
 
+            terminated[agent] = bool(terminated[agent])
+
         return terminated
 
     @override

diff --git a/momadm_benchmarks/envs/crazyrl/escort/escort.py b/momadm_benchmarks/envs/crazyrl/escort/escort.py
@@ -258,6 +258,8 @@ def _compute_terminated(self):
                     terminated[other_agent] = True
                 self.agents = []
 
+            terminated[agent] = bool(terminated[agent])
+
         return terminated
 
     @override

diff --git a/momadm_benchmarks/envs/crazyrl/surround/surround.py b/momadm_benchmarks/envs/crazyrl/surround/surround.py
@@ -236,6 +236,8 @@ def _compute_terminated(self):
                     terminated[other_agent] = True
                 self.agents = []
 
+            terminated[agent] = bool(terminated[agent])
+
         return terminated
 
     @override