I've had issues with both the Jupyter notebook (https://github.com/drillan/jupyter-black) and lab (https://github.com/ryantam626/jupyterlab_code_formatter) extensions, but would be good to figure this out.