Hello All, 

I just tried running the following notebook (https://docs.databricks.com/aws/en/notebooks/source/machine-learning/optuna-mlflow.html)  on the Databricks Free Edition platform , through Microsoft Account Authentication. It takes 15 minutes to run 8 trials then outputs a grpc_status:13 error before it breaks. Could this be because the distributed MlflowSparkStudy class cannot work on  the Free Edition, or is this an internal bug.

Thank you in advance !