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

fix merge thingy

This commit is contained in:
Cadene 2024-04-16 17:24:25 +00:00
parent 36d9e885ef
commit 4327e43f19
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
download_and_upload_dataset.py
View File

@ -61,7 +61,7 @@ def download_and_upload_pusht(root, root_tests, dataset_id="pusht", fps=10):
import pymunk import pymunk
from gym_pusht.envs.pusht import PushTEnv, pymunk_to_shapely from gym_pusht.envs.pusht import PushTEnv, pymunk_to_shapely
from lerobot.common.policies.diffusion.replay_buffer import ( from lerobot.common.datasets._diffusion_policy_replay_buffer import (
ReplayBuffer as DiffusionPolicyReplayBuffer, ReplayBuffer as DiffusionPolicyReplayBuffer,
) )
except ModuleNotFoundError as e: except ModuleNotFoundError as e:

2
lerobot/common/datasets/_diffusion_policy_replay_buffer.py
View File

@ -1,6 +1,6 @@
"""Helper code for loading PushT dataset from Diffusion Policy (https://diffusion-policy.cs.columbia.edu/) """Helper code for loading PushT dataset from Diffusion Policy (https://diffusion-policy.cs.columbia.edu/)
Copied from the original Diffusion Policy repository. Copied from the original Diffusion Policy repository and used in our `download_and_upload_dataset.py` script.
""" """
from __future__ import annotations from __future__ import annotations