fix: fix bugs of GlobalEgoState

Emerge-Lab · Dec 15, 2024 · 35a5c78 · 35a5c78
1 parent 7ada291
commit 35a5c78
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pygpudrive/datatypes/observation.py b/pygpudrive/datatypes/observation.py
@@ -90,7 +90,7 @@ def __init__(self, abs_self_obs_tensor: torch.Tensor):
         self.pos_x = abs_self_obs_tensor[:, :, 0]
         self.pos_y = abs_self_obs_tensor[:, :, 1]
         self.pos_z = abs_self_obs_tensor[:, :, 2]
-        self.rotation_as_quaternion = abs_self_obs_tensor[:, :, 3:6]
+        self.rotation_as_quaternion = abs_self_obs_tensor[:, :, 3:7]
         self.rotation_angle = abs_self_obs_tensor[:, :, 7]
         self.goal_x = abs_self_obs_tensor[:, :, 8]
         self.goal_y = abs_self_obs_tensor[:, :, 9]

diff --git a/src/sim.cpp b/src/sim.cpp
@@ -665,6 +665,7 @@ inline void collectAbsoluteObservationsSystem(Engine &ctx,
     out.rotation.rotationFromAxis = utils::quatToYaw(rotation);
     out.goal = goal;
     out.vehicle_size = vehicleSize;
+    out.id = ctx.get<AgentID>(agent_iface.e).id;
 }
 
 void setupRestOfTasks(TaskGraphBuilder &builder, const Sim::Config &cfg,