From 84a1647c012e972cb453bc715b76fb6d0441190b Mon Sep 17 00:00:00 2001 From: Cadene Date: Mon, 11 Mar 2024 12:41:14 +0000 Subject: [PATCH] fix import --- lerobot/common/datasets/abstract.py | 3 +-- lerobot/common/envs/factory.py | 4 +--- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/lerobot/common/datasets/abstract.py b/lerobot/common/datasets/abstract.py index 0c0746e1..d6a51246 100644 --- a/lerobot/common/datasets/abstract.py +++ b/lerobot/common/datasets/abstract.py @@ -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.storages import TensorStorage, _collate_id from torchrl.data.replay_buffers.writers import ImmutableDatasetWriter, Writer - -from rl.torchrl.envs.transforms.transforms import Compose +from torchrl.envs.transforms.transforms import Compose class AbstractExperienceReplay(TensorDictReplayBuffer): diff --git a/lerobot/common/envs/factory.py b/lerobot/common/envs/factory.py index 184646cf..c20ef441 100644 --- a/lerobot/common/envs/factory.py +++ b/lerobot/common/envs/factory.py @@ -1,6 +1,4 @@ -from torchrl.envs.transforms import StepCounter, TransformedEnv - -from rl.torchrl.envs.transforms.transforms import Compose, Transform +from torchrl.envs.transforms import Compose, StepCounter, Transform, TransformedEnv def make_env(cfg, transform=None):