From 253565da477e76124271198f53b9e358f6961e47 Mon Sep 17 00:00:00 2001 From: vmoens Date: Tue, 20 Feb 2024 16:25:17 -0800 Subject: [PATCH] fix --- 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 e1b2fac35f7..82d4a9b5090 100644 --- a/torchrl/data/replay_buffers/storages.py +++ b/torchrl/data/replay_buffers/storages.py @@ -147,7 +147,7 @@ def shape(self): f"Please report this exception as well as the use case (incl. buffer construction) on github." ) - def _max_size_along_dim0(self, *, single_data, batched_data): + def _max_size_along_dim0(self, *, single_data=None, batched_data=None): if self.ndim == 1: return self.max_size raise RuntimeError(