fix

pytorch · Feb 21, 2024 · 253565d · 253565d
1 parent b049c49
commit 253565d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(