Updates for batch-partition example

Reflecting latest changes for core packages.