rename to image_transforms

2024-06-06 16:50:22 +00:00
parent a86f387554
commit c45dd8f848
8 changed files with 23 additions and 24 deletions
--- a/tests/scripts/save_image_transforms.py
+++ b/tests/scripts/save_image_transforms.py
@@ -15,7 +15,7 @@ to_pil = v2.ToPILImage()


 def main(repo_id):
-    dataset = LeRobotDataset(repo_id, transform=None)
+    dataset = LeRobotDataset(repo_id, image_transforms=None)
    output_dir = Path(ARTIFACT_DIR)
    output_dir.mkdir(parents=True, exist_ok=True)