From c80f9da303e85cab264b815add7ee50b6e911511 Mon Sep 17 00:00:00 2001 From: Callum Tilbury Date: Fri, 14 Jun 2024 16:55:26 +0200 Subject: [PATCH] fix: add item buffer to init. --- flashbax/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/flashbax/__init__.py b/flashbax/__init__.py index f685c37..576f501 100644 --- a/flashbax/__init__.py +++ b/flashbax/__init__.py @@ -19,10 +19,12 @@ make_flat_buffer, make_item_buffer, make_prioritised_flat_buffer, + make_prioritised_item_buffer, make_prioritised_trajectory_buffer, make_trajectory_buffer, make_trajectory_queue, prioritised_flat_buffer, + prioritised_item_buffer, prioritised_trajectory_buffer, trajectory_buffer, trajectory_queue,