amend

pytorch · Jan 9, 2024 · 7caddc3 · 7caddc3
1 parent 60a74d6
commit 7caddc3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/torchrl/data/datasets/d4rl.py b/torchrl/data/datasets/d4rl.py
@@ -24,7 +24,7 @@
 from torchrl.data.datasets.d4rl_infos import D4RL_DATASETS
 
 from torchrl.data.datasets.utils import _get_root_dir
-from torchrl.data.replay_buffers import TensorDictReplayBuffer
+from torchrl.data.replay_buffers.replay_buffers import TensorDictReplayBuffer
 from torchrl.data.replay_buffers.samplers import Sampler
 from torchrl.data.replay_buffers.storages import TensorStorage
 from torchrl.data.replay_buffers.writers import ImmutableDatasetWriter, Writer

diff --git a/torchrl/data/datasets/openml.py b/torchrl/data/datasets/openml.py
@@ -9,12 +9,12 @@
 import numpy as np
 from tensordict.tensordict import TensorDict
 
-from torchrl.data import TensorStorage
 from torchrl.data.datasets.utils import _get_root_dir
 from torchrl.data.replay_buffers import (
     Sampler,
     SamplerWithoutReplacement,
     TensorDictReplayBuffer,
+    TensorStorage,
     Writer,
 )