Skip to content

Commit a717121

Browse files
committed
cache _ALL_INDICES to avoid reallocations
1 parent b0f90d6 commit a717121

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

source/isaaclab/isaaclab/assets/articulation/multirotor.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -119,9 +119,9 @@ def reset(self, env_ids: Sequence[int] | None = None):
119119

120120
# reset multirotor-specific data
121121
if env_ids is None:
122-
env_ids = torch.arange(self.num_instances, device=self.device)
123-
else:
124-
env_ids = torch.tensor(env_ids, device=self.device)
122+
env_ids = self._ALL_INDICES
123+
elif not isinstance(env_ids, torch.Tensor):
124+
env_ids = torch.tensor(env_ids, dtype=torch.long, device=self.device)
125125

126126
# reset thruster targets to default values
127127
if self._data.thrust_target is not None and self._data.default_thruster_rps is not None:

0 commit comments

Comments
 (0)