wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

revert draft

This commit is contained in:
Cadene 2024-04-22 15:00:18 +00:00
parent db9ac59230
commit b138f114c8
1 changed files with 0 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
lerobot/common/envs/factory.py
View File

@ -40,13 +40,4 @@ def make_env(cfg, num_parallel_envs=0) -> gym.Env | gym.vector.SyncVectorEnv:
]
)
def preprocess_observation(observation):
return observation / 255.0
def preprocess_reward(reward):
return reward / 255.0
env = gym.wrappers.TransformObservation(env, preprocess_observation)
env = gym.wrappers.TransformReward(env, preprocess_reward)
return env