wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lerobot/scripts/server/buffer.py
View File

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

6
lerobot/scripts/server/maniskill_manipulator.py
View File

@ -1,5 +1,3 @@
import logging
import time
from typing import Any from typing import Any
import einops import einops
@ -10,10 +8,6 @@ from mani_skill.utils.wrappers.record import RecordEpisode
from mani_skill.vector.wrappers.gymnasium import ManiSkillVectorEnv from mani_skill.vector.wrappers.gymnasium import ManiSkillVectorEnv
from lerobot.common.envs.configs import ManiskillEnvConfig 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( def preprocess_maniskill_observation(