Merge branch 'curobo' of https://github.com/StanfordVL/OmniGibson into curobo
This commit is contained in:
commit
eabc9f21d4
|
@ -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))
|
||||
|
|
Loading…
Reference in New Issue