diff --git a/src/lerobot/lerobot_types.py b/src/lerobot/lerobot_types.py index a05b27712..9de504870 100644 --- a/src/lerobot/lerobot_types.py +++ b/src/lerobot/lerobot_types.py @@ -23,7 +23,7 @@ import numpy as np import torch -class TransitionKey(str, Enum): # noqa: UP042 +class TransitionKey(str, Enum): """Keys for accessing EnvTransition dictionary components.""" # TODO(Steven): Use consts