Skip to content

Commit 5b99208

Browse files
Reduce code and rely on device checking in tensor.to
Co-authored-by: Mayank Mittal <12863862+Mayankm96@users.noreply.github.com> Signed-off-by: jtigue-bdai <166445701+jtigue-bdai@users.noreply.github.com>
1 parent 14bde30 commit 5b99208

File tree

1 file changed

+1
-4
lines changed
  • source/extensions/omni.isaac.lab/omni/isaac/lab/utils/noise

1 file changed

+1
-4
lines changed

source/extensions/omni.isaac.lab/omni/isaac/lab/utils/noise/noise_model.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -30,10 +30,7 @@ def constant_noise(data: torch.Tensor, cfg: noise_cfg.ConstantNoiseCfg) -> torch
3030

3131
# fix tensor device for bias on first call and update config parameters
3232
if isinstance(cfg.bias, torch.Tensor):
33-
if cfg.bias.device is not data.device:
34-
cfg.bias = cfg.bias.to(
35-
data.device,
36-
)
33+
cfg.bias = cfg.bias.to(device=data.device)
3734

3835
if cfg.operation == "add":
3936
return data + cfg.bias

0 commit comments

Comments
 (0)