diff --git a/lerobot/common/datasets/factory.py b/lerobot/common/datasets/factory.py index 7e0794761..4ae161f69 100644 --- a/lerobot/common/datasets/factory.py +++ b/lerobot/common/datasets/factory.py @@ -74,7 +74,6 @@ def make_dataset( transforms = v2.Compose( [ - # TODO(rcadene): we need to do something about image_keys Prod(in_keys=clsfunc.image_keys, prod=1 / 255.0), NormalizeTransform( stats,