This commit is contained in:
Cadene 2024-05-04 10:55:17 +00:00
parent 8883f3b28a
commit 7b3cf7e08c
1 changed files with 1 additions and 1 deletions

View File

@ -15,7 +15,7 @@ from lerobot.common.datasets.utils import (
from lerobot.common.datasets.video_utils import VideoFrame, load_from_videos from lerobot.common.datasets.video_utils import VideoFrame, load_from_videos
DATA_DIR = Path(os.environ["DATA_DIR"]) if "DATA_DIR" in os.environ else None DATA_DIR = Path(os.environ["DATA_DIR"]) if "DATA_DIR" in os.environ else None
CODEBASE_VERSION = "v1.2" CODEBASE_VERSION = "v1.3"
class LeRobotDataset(torch.utils.data.Dataset): class LeRobotDataset(torch.utils.data.Dataset):