tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix import

Browse Source
This commit is contained in:
Cadene
2024-03-11 12:41:14 +00:00
parent 2a4bc7371e
commit ebd5c786f1
2 changed files with 2 additions and 5 deletions
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):