-
Notifications
You must be signed in to change notification settings - Fork 0
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Gpr dataset integration #8
Conversation
|
||
def check_format(raw_dir) -> bool: | ||
"""Verify HDF5 files have expected structure""" | ||
print(f"[DEBUG] Checking format for directory: {raw_dir}") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
logging.debug everywhere please
|
||
if "next.success" in batch: | ||
rr.log("next.success", rr.Scalar(batch["next.success"][i].item())) | ||
# Choose visualization function based on dataset type |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
remove that please and add more ifs if necessary
file_type = get_krec_file_type(file_path) | ||
|
||
if file_type == "krec": | ||
return load_krec_direct(file_path) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
return load_krec_direct(file_path) | |
return krec.KRec.load(krec_file_path) |
simplicity always wins, no langchain patterns
…ot into gpr_dataset_integration
Integrate GPR h5 data to be used via LeRobot Dataset
First clone the repo and do the install instructions in readme
then just run
To visualize: