[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-04-04 07:59:22 +00:00
parent e86fe66dbd
commit 037ecae9e0
2 changed files with 1 additions and 8 deletions

View File

@ -15,7 +15,6 @@
# limitations under the License.
import functools
import io
import os
import pickle
from typing import Any, Callable, Optional, Sequence, TypedDict
@ -388,8 +387,8 @@ class ReplayBuffer:
Yields:
BatchTransition: Prefetched batch transitions
"""
import threading
import queue
import threading
# Use thread-safe queue
data_queue = queue.Queue(maxsize=queue_size)

View File

@ -1,5 +1,3 @@
import logging
import time
from typing import Any
import einops
@ -10,10 +8,6 @@ from mani_skill.utils.wrappers.record import RecordEpisode
from mani_skill.vector.wrappers.gymnasium import ManiSkillVectorEnv
from lerobot.common.envs.configs import ManiskillEnvConfig
from lerobot.configs import parser
from lerobot.configs.train import TrainPipelineConfig
from lerobot.common.policies.sac.configuration_sac import SACConfig
from lerobot.common.policies.sac.modeling_sac import SACPolicy
def preprocess_maniskill_observation(