Jump to function definition not working after cell with function definition have been executed

[PeterFogh]

Hi, first of all, thanks for this nice extension.

My problem is that jump to function definition does not work after the cell with the function definition have been executed.

My configuration

conda package: Jupyterlab=2.1.5
Jupyter extension: @krassowski/[email protected]

more specific:
conda create -n py37_jupyterlab_v2.1.5 python=3.7 jupyterlab nb_conda_kernels nodejs nbresuse flake8 jupyterlab_code_formatter black ipympl==0.5.6 nbdime dask-labextension jupyterlab-git pandas holoviews
and
conda activate py37_jupyterlab_v2.1.5; jupyter labextension install @ijmbarr/[email protected] @jupyter-widgets/[email protected] @jupyterlab/[email protected] @jupyterlab/[email protected] @jupyterlab/[email protected] @jupyterlab/[email protected] @jupyterlab/[email protected] @krassowski/[email protected] @pyviz/[email protected] @ryantam626/[email protected] [email protected] [email protected] [email protected] [email protected] --no-build; jupyter lab build; jupyter serverextension enable --py jupyterlab_code_formatter; jupyter serverextension enable dask_labextension

Reproduce bug

1. Open a new Jupyter Notebook (without executing any cells)
2. Insert this code in a cell:

   def test_go_to_difinition(arg1):
       """Test function for the go to definition."""
       pass
   
   test_go_to_difinition(1)
   

3. ALT+click on test_go_to_difinition(1) - this should work and the cursor jumps to the definition of the function.
4. Now, execute the cell, and the ALT+click on test_go_to_difinition(1) does not make the cursor jumps to the definition of the function

[poxzlm]

@krassowski @PeterFogh
I like this plugin!
Has this problem been resolved?