I'm getting an error

You haven’t configured the CLI yet! Please configure by entering `/databricks/python_shell/scripts/db_ipykernel_launcher.py configure`

My cluster is running Databricks Runtime Version 10.1

I've also installed mlflow to the cluster libraries, I've tried just using the runtime without installing but couldn't import mlflow

I found this question , but the solution there didn't help