Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature] GenDGRL #1773

Merged
merged 14 commits into from
Jan 16, 2024
Prev Previous commit
Next Next commit
amend
  • Loading branch information
vmoens committed Jan 8, 2024
commit 44d1f9a0ab24fe0db43da7c23f2d253715e6a40a
3 changes: 2 additions & 1 deletion torchrl/data/datasets/gen_dgrl.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,8 @@
import torch

from tensordict import TensorDict
from torchrl.data import TensorDictReplayBuffer, TensorStorage
from torchrl.data.replay_buffers.replay_buffers import TensorDictReplayBuffer
from torchrl.data.replay_buffers.storages import TensorStorage
from torchrl.data.datasets.utils import _get_root_dir
from torchrl.envs.utils import _classproperty

Expand Down
Loading