For information, upgrading python libraries does not resolve all problems.

This code works fine on databricks in a notebook :

import mlflow
model = mlflow.spark.load_model('runs:/cb6ff62587a0404cabeadd47e4c9408a/model')

Whereas it failed on intelliJ with databricks-connect

Did you have any solution ?