Fixed missing 'batch' method in __init__.py