Skip to content

Commit 373bae1

Browse files
authored
Merge pull request #1 from YilingQiao/Roushelfy/ipc-coupling-squashed
format
2 parents 239c60a + 562c3fc commit 373bae1

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

genesis/engine/couplers/ipc_coupler.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1019,7 +1019,9 @@ def _set_genesis_transforms_from_ipc(self, abd_data_by_link):
10191019

10201020
# Convert to torch tensors and add batch dimension
10211021
pos_tensor = torch.as_tensor(pos, dtype=gs.tc_float, device=gs.device).unsqueeze(0) # (1, 3)
1022-
quat_tensor = torch.as_tensor(quat_wxyz, dtype=gs.tc_float, device=gs.device).unsqueeze(0) # (1, 4) [w, x, y, z]
1022+
quat_tensor = torch.as_tensor(quat_wxyz, dtype=gs.tc_float, device=gs.device).unsqueeze(
1023+
0
1024+
) # (1, 4) [w, x, y, z]
10231025

10241026
# Determine if this is a base link
10251027
is_base_link = (link_idx == entity.base_link_idx)
@@ -1066,7 +1068,9 @@ def _set_genesis_transforms_from_ipc(self, abd_data_by_link):
10661068
entity.zero_all_dofs_velocity(envs_idx=None, unsafe=True)
10671069

10681070
except Exception as e:
1069-
gs.logger.warning(f"Failed to set Genesis transform for IPC-only link {link_idx}, env {env_idx}: {e}")
1071+
gs.logger.warning(
1072+
f"Failed to set Genesis transform for IPC-only link {link_idx}, env {env_idx}: {e}"
1073+
)
10701074
continue
10711075

10721076
def _get_genesis_link_transform(self, link_idx, env_idx):

0 commit comments

Comments
 (0)