Merge branch 'curobo' of https://github.com/StanfordVL/OmniGibson into curobo

This commit is contained in:
Frank Yang 2024-10-01 00:23:21 -07:00
commit eabc9f21d4
1 changed files with 1 additions and 0 deletions

View File

@ -391,6 +391,7 @@ class RigidContactAPIImpl:
if not th.any(interesting_impulses > 0):
return set()
# Get all of the (row, col) pairs where the impulse is greater than 0
return {
(interesting_row_paths[row], interesting_col_paths[col])
for row, col in zip(*th.nonzero(interesting_impulses > 0, as_tuple=True))