Batch size finder code example in dark mode is light instead of dark

Batch size finder code example in dark mode is light instead of dark #29569