Hi,
the following code is intended to calculate the orientation from current robot pose to the target in the x-y plane.
So the first two indices should be 1 instead of 2.
walk_target_angle = torch.atan2(targets[:, 2] - torso_positions[:, 2], targets[:, 0] - torso_positions[:, 0])