Скалярные значения по умолчанию работают так, как вы ожидаете, и batch_size: int = 500 — это все, что вам нужно.