diff --git a/lerobot/scripts/eval.py b/lerobot/scripts/eval.py index a1d5fc24..2ad3b36e 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, infer_envs_task, preprocess_observation +from lerobot.common.envs.utils import 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