Skip to content

Commit 33887fc

Browse files
ZaberKoMilotrince
authored andcommitted
Fix pickling of 'reward_cfg["reward_scales"]' in hover example (Genesis-Embodied-AI#1340)
1 parent ad256ed commit 33887fc

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

examples/drone/hover_env.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import torch
22
import math
3+
import copy
34
import genesis as gs
45
from genesis.utils.geom import quat_to_xyz, transform_by_quat, inv_quat, transform_quat_by_quat
56

@@ -28,7 +29,7 @@ def __init__(self, num_envs, env_cfg, obs_cfg, reward_cfg, command_cfg, show_vie
2829
self.command_cfg = command_cfg
2930

3031
self.obs_scales = obs_cfg["obs_scales"]
31-
self.reward_scales = reward_cfg["reward_scales"]
32+
self.reward_scales = copy.deepcopy(reward_cfg["reward_scales"])
3233

3334
# create scene
3435
self.scene = gs.Scene(

0 commit comments

Comments
 (0)