From 22a84e640b2a9cb639d8be4b34608b6faf6e2c70 Mon Sep 17 00:00:00 2001 From: Nikhil Rao Date: Wed, 12 Oct 2022 17:53:29 -0700 Subject: [PATCH] Displaying keys in alphabetical order during replay buffer init --- torchrl/data/replay_buffers/storages.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchrl/data/replay_buffers/storages.py b/torchrl/data/replay_buffers/storages.py index daf1d8d623a..e8967ab0390 100644 --- a/torchrl/data/replay_buffers/storages.py +++ b/torchrl/data/replay_buffers/storages.py @@ -204,7 +204,7 @@ def _init(self, data: Union[TensorDictBase, torch.Tensor]) -> None: else: out = TensorDict({}, [self.max_size, *data.shape]) print("The storage is being created: ") - for key, tensor in data.items(): + for key, tensor in sorted(data.items()): if isinstance(tensor, TensorDictBase): out[key] = ( tensor.expand(self.max_size)