diff --git a/lerobot/common/datasets/factory.py b/lerobot/common/datasets/factory.py index 8407c168..07afb614 100644 --- a/lerobot/common/datasets/factory.py +++ b/lerobot/common/datasets/factory.py @@ -70,7 +70,7 @@ def make_dataset( transform=Prod(in_keys=clsfunc.image_keys, prod=1 / 255.0), ) stats = compute_stats(stats_dataset) - os.makedirs(precomputed_stats_path.parent, exist_ok=True) + precomputed_stats_path.parent.mkdir(parents=True, exist_ok=True) torch.save(stats, precomputed_stats_path) else: stats = torch.load(stats_path)