Skip to content

Commit

Permalink
fix bug in generating target_frame_body_names
Browse files Browse the repository at this point in the history
Signed-off-by: Tao <[email protected]>
  • Loading branch information
taochenshh authored Dec 21, 2023
1 parent 16b0b99 commit 6a814ba
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@

from __future__ import annotations

import re
import torch
from typing import TYPE_CHECKING, Sequence

Expand Down Expand Up @@ -230,7 +231,7 @@ def _initialize_impl(self):
first_env_prim_paths = all_prim_paths[0 : self._num_target_body_frames + 1]
first_env_body_names = [first_env_prim_path.split("/")[-1] for first_env_prim_path in first_env_prim_paths]

target_frame_body_names = first_env_body_names[1:]
target_frame_body_names = [x for x in first_env_body_names if not re.match(self._tracked_body_names[0], x)]

# The position and rotation components of target frame offsets
target_frame_offset_pos = []
Expand Down

0 comments on commit 6a814ba

Please sign in to comment.