python – WSL:Ubuntu in Visual studio code – losing connection

I set up visual studio code to code with python using Windows 11.
I wanted to use a package for python called sage (math software), but to use it with windows I need to connect to WSL:Ubuntu.

Now, everytime I start vsc it successfully connects to Ubunute, but loses connection after 1 or 2 minutes. A window pops up saying it tries to reconnect, but this is never succesful. Only restarting helps.
Since I need to execute a program which takes more than a minute, this is a problem for me.

For the remote-connection to Ubuntu I am using the extension WSL v0.104.3, which is the newest one I could find. It also says that there are no updates for VSC.

Is anyone familiar with this problem?

Thanks in advance.