diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py index e6300dbd..d6f1105b 100644 --- a/lerobot/scripts/eval.py +++ b/lerobot/scripts/eval.py @@ -66,7 +66,7 @@ from torch import Tensor, nn from tqdm import trange from lerobot.common.envs.factory import make_env -from lerobot.common.envs.utils import check_env_attributes_and_types, preprocess_observation, add_envs_task +from lerobot.common.envs.utils import add_envs_task, check_env_attributes_and_types, preprocess_observation from lerobot.common.policies.factory import make_policy from lerobot.common.policies.pretrained import PreTrainedPolicy from lerobot.common.policies.utils import get_device_from_parameters