fix(builder): fix bug in builder (#75)

PKU-Alignment · Aug 22, 2023 · 9823134 · 9823134
1 parent f499491
commit 9823134
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/safety_gymnasium/builder.py b/safety_gymnasium/builder.py
@@ -246,11 +246,7 @@ def step(self, action: np.ndarray) -> tuple[np.ndarray, float, float, bool, bool
 
         if self.render_parameters.mode == 'human':
             self.render()
-
-        terminateds = {'agent_0': self.terminated, 'agent_1': self.terminated}
-        truncateds = {'agent_0': self.truncated, 'agent_1': self.truncated}
-
-        return self.task.obs(), reward, cost, terminateds, truncateds, info
+        return self.task.obs(), reward, cost, self.terminated, self.truncated, info
 
     def _reward(self) -> float:
         """Calculate the current rewards.