From 918d1fc5cac69ffb56d1a5f9ac098b864121d5bd Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Sat, 14 Dec 2024 19:22:04 -0800 Subject: [PATCH] Update [ghstack-poisoned] --- torchrl/data/replay_buffers/replay_buffers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchrl/data/replay_buffers/replay_buffers.py b/torchrl/data/replay_buffers/replay_buffers.py index 67113095af0..f9e76a2a282 100644 --- a/torchrl/data/replay_buffers/replay_buffers.py +++ b/torchrl/data/replay_buffers/replay_buffers.py @@ -1159,7 +1159,7 @@ class TensorDictReplayBuffer(ReplayBuffer): def __init__(self, *, priority_key: str = "td_error", **kwargs) -> None: writer = kwargs.get("writer", None) if writer is None: - kwargs["writer"] = TensorDictRoundRobinWriter() + kwargs["writer"] = TensorDictRoundRobinWriter(compilable=kwargs.get("compilable")) super().__init__(**kwargs) self.priority_key = priority_key