I'm attempting to switch Pylance to use a pull method for diagnostics but I'm not getting textDocument/diagnostic messages for notebook cells.

I believe that's because notebook changes aren't tracked here:

And notebook opens aren't tracked here: