Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion roboverse_pack/callback_funcs/humanoid/reset_funcs.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ def random_root_state(
env.setup_initial_env_states.robots[env.name].root_state[env_ids, 0:3] = positions
env.setup_initial_env_states.robots[env.name].root_state[env_ids, 3:7] = orientations
env.setup_initial_env_states.robots[env.name].root_state[env_ids, 7:13] = velocities
# # set into the physics simulation
# set into the physics simulation
# env.write_robot_root_state(torch.cat([positions, orientations, velocities], dim=-1), env_ids=env_ids)


Expand Down
2 changes: 1 addition & 1 deletion roboverse_pack/callback_funcs/humanoid/reward_funcs.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@

def track_lin_vel_xy(env: EnvTypes, env_states: TensorState, std: float) -> torch.Tensor:
"""Reward tracking of linear velocity commands (xy axes) in the gravity aligned robot frame using exponential kernel."""
# extract the used quantities (to enable type-hinting)
# extract the used quantities (to enable type-hinting).
robot_state = env_states.robots[env.name]
base_quat = robot_state.root_state[:, 3:7]
base_lin_vel = quat_rotate_inverse(base_quat, robot_state.root_state[:, 7:10])
Expand Down