スカラーのデフォルトは期待通りに機能し、batch_size: int = 500 があれば十分です。