From 3b6fd7b19c2ce55e0123aff2528f91b7b7ba4f92 Mon Sep 17 00:00:00 2001 From: Edan Toledo <42650996+EdanToledo@users.noreply.github.com> Date: Thu, 14 Dec 2023 10:15:59 +0000 Subject: [PATCH] Update flashbax/buffers/item_buffer.py Co-authored-by: Callum Tilbury <37700709+callumtilbury@users.noreply.github.com> --- flashbax/buffers/item_buffer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flashbax/buffers/item_buffer.py b/flashbax/buffers/item_buffer.py index e33614d..de30a15 100644 --- a/flashbax/buffers/item_buffer.py +++ b/flashbax/buffers/item_buffer.py @@ -61,7 +61,7 @@ def create_item_buffer( min_length (int): The minimum length of the buffer. sample_batch_size (int): The batch size of the samples. add_sequences (Optional[bool], optional): Whether data is being added in sequences - to the buffer. If False, single transitions are being added each time add + to the buffer. If False, single items are being added each time add is called. Defaults to False. add_batches: (Optional[bool], optional): Whether adding data in batches to the buffer. If False, single transitions or single sequences are being added each time add