I also recommend looking at ipyannotate if most of your workflow is in jupyter