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

[nicolasperez19]

📚 Documentation

On the Batch size finder advanced tricks page, in dark mode the example code is in light mode makes it hard to read:

The code example should look instead look like this in dark mode:

cc @lantiga @Borda