Update

[ghstack-poisoned]
pytorch · Dec 15, 2024 · 795a390 · 795a390
2 parents 056e6a4 + 0de0641
commit 795a390
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchrl/data/replay_buffers/writers.py b/torchrl/data/replay_buffers/writers.py
@@ -332,7 +332,7 @@ def extend(self, data: Sequence) -> torch.Tensor:
         # Other than that, a "flat" (1d) index is ok to write the data
         self._storage.set(index, data)
         index = self._replicate_index(index)
-        for ent in self._storage._attached_entities:
+        for ent in list(self._storage._attached_entities):
             ent.mark_update(index)
         return index