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

remove hardcoding

This commit is contained in:
Remi Cadene 2024-05-23 09:14:17 +00:00
parent a7d24f6dc2
commit 10c5151fc6
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lerobot/common/datasets/push_dataset_to_hub/aloha_dora_format.py
View File

@ -32,8 +32,6 @@ from lerobot.common.utils.utils import init_logging
def check_format(raw_dir) -> bool: def check_format(raw_dir) -> bool:
# TODO(rcadene): remove hardcoding
raw_dir = raw_dir / "018fa076-ad19-7c77-afa4-49f7f072e86f"
assert raw_dir.exists() assert raw_dir.exists()
leader_file = list(raw_dir.glob("*.parquet")) leader_file = list(raw_dir.glob("*.parquet"))
@ -43,9 +41,6 @@ def check_format(raw_dir) -> bool:
def load_from_raw(raw_dir: Path, out_dir: Path): def load_from_raw(raw_dir: Path, out_dir: Path):
# TODO(rcadene): remove hardcoding
raw_dir = raw_dir / "018fa076-ad19-7c77-afa4-49f7f072e86f"
# Load data stream that will be used as reference for the timestamps synchronization # Load data stream that will be used as reference for the timestamps synchronization
reference_key = "observation.images.cam_right_wrist" reference_key = "observation.images.cam_right_wrist"
reference_df = pd.read_parquet(raw_dir / f"{reference_key}.parquet") reference_df = pd.read_parquet(raw_dir / f"{reference_key}.parquet")