I have the same issue, did you manage to solve it?
I have the Databricks extension well configured and my role has enough permissions. Everything used to work propertly but now when I run my notebooks is giving me this issue and 'no module named dbruntime' all the time.