fix: add item buffer to init.

instadeepai · Jun 14, 2024 · c80f9da · c80f9da
1 parent cd76785
commit c80f9da
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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,