diff --git a/src/saturn_engine/worker/inventories/__init__.py b/src/saturn_engine/worker/inventories/__init__.py index e50049d0..6183551c 100644 --- a/src/saturn_engine/worker/inventories/__init__.py +++ b/src/saturn_engine/worker/inventories/__init__.py @@ -4,6 +4,7 @@ from ..inventory import Inventory from ..inventory import Item from ..inventory import IteratorInventory +from .batching import BatchingInventory from .chained import ChainedInventory from .dummy import DummyInventory from .joined import JoinedInventory @@ -18,4 +19,5 @@ "JoinedInventory": JoinedInventory, "ChainedInventory": ChainedInventory, "PeriodicInventory": PeriodicInventory, + "BatchingInventory": BatchingInventory, }