Add dataset visualization with rerun.io (#131)

Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>

lerobot/common/datasets/lerobot_dataset.py

@@ -1,3 +1,4 @@
+import os
 from pathlib import Path
 
 import datasets
@@ -13,7 +14,8 @@ from lerobot.common.datasets.utils import (
 )
 from lerobot.common.datasets.video_utils import VideoFrame, load_from_videos
 
-CODEBASE_VERSION = "v1.2"
+DATA_DIR = Path(os.environ["DATA_DIR"]) if "DATA_DIR" in os.environ else None
+CODEBASE_VERSION = "v1.3"
 
 
 class LeRobotDataset(torch.utils.data.Dataset):
@@ -21,7 +23,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
         self,
         repo_id: str,
         version: str | None = CODEBASE_VERSION,
-        root: Path | None = None,
+        root: Path | None = DATA_DIR,
         split: str = "train",
         transform: callable = None,
         delta_timestamps: dict[list[float]] | None = None,