[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
e86fe66dbd
commit
037ecae9e0
|
@ -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)
|
||||
|
|
|
@ -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(
|
||||
|
|
Loading…
Reference in New Issue