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

fix import

This commit is contained in:
Cadene 2024-03-11 12:41:14 +00:00
parent ccd5dc5a42
commit 84a1647c01
2 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lerobot/common/datasets/abstract.py
View File

@ -13,8 +13,7 @@ from torchrl.data.replay_buffers.replay_buffers import TensorDictReplayBuffer
from torchrl.data.replay_buffers.samplers import SliceSampler from torchrl.data.replay_buffers.samplers import SliceSampler
from torchrl.data.replay_buffers.storages import TensorStorage, _collate_id from torchrl.data.replay_buffers.storages import TensorStorage, _collate_id
from torchrl.data.replay_buffers.writers import ImmutableDatasetWriter, Writer from torchrl.data.replay_buffers.writers import ImmutableDatasetWriter, Writer
from torchrl.envs.transforms.transforms import Compose
from rl.torchrl.envs.transforms.transforms import Compose
class AbstractExperienceReplay(TensorDictReplayBuffer): class AbstractExperienceReplay(TensorDictReplayBuffer):

4
lerobot/common/envs/factory.py
View File

@ -1,6 +1,4 @@
from torchrl.envs.transforms import StepCounter, TransformedEnv from torchrl.envs.transforms import Compose, StepCounter, Transform, TransformedEnv
from rl.torchrl.envs.transforms.transforms import Compose, Transform
def make_env(cfg, transform=None): def make_env(cfg, transform=None):