diff --git a/src/lerobot/envs/libero.py b/src/lerobot/envs/libero.py index 0672f583f..5d0d37c83 100644 --- a/src/lerobot/envs/libero.py +++ b/src/lerobot/envs/libero.py @@ -4,10 +4,9 @@ import logging import math import os from collections import defaultdict -from collections.abc import Callable -from itertools import chain -from typing import Any, Dict, List from collections.abc import Callable, Iterable, Mapping, Sequence +from itertools import chain +from typing import Any import gymnasium as gym import numpy as np diff --git a/src/lerobot/policies/smolvla/modeling_smolvla.py b/src/lerobot/policies/smolvla/modeling_smolvla.py index 6bf956aa3..18f2fc58a 100644 --- a/src/lerobot/policies/smolvla/modeling_smolvla.py +++ b/src/lerobot/policies/smolvla/modeling_smolvla.py @@ -938,4 +938,4 @@ class VLAFlowMatching(nn.Module): suffix_out = suffix_out[:, -self.config.chunk_size :] suffix_out = suffix_out.to(dtype=torch.float32) v_t = self.action_out_proj(suffix_out) - return v_t \ No newline at end of file + return v_t diff --git a/src/lerobot/scripts/eval.py b/src/lerobot/scripts/eval.py index bd131f708..4777f57e3 100644 --- a/src/lerobot/scripts/eval.py +++ b/src/lerobot/scripts/eval.py @@ -52,14 +52,13 @@ import logging import threading import time from collections import defaultdict -from collections.abc import Callable +from collections.abc import Callable, Iterator from contextlib import nullcontext from copy import deepcopy from dataclasses import asdict from pathlib import Path from pprint import pformat -from typing import Dict, List, Tuple, TypedDict -from collections.abc import Iterator +from typing import TypedDict import einops import gymnasium as gym