MLflow: Connect python with Community Edition

[nicobuko]

Hello,

I am new to databricks and want to work with MLFlow in the Databricks Community Edition. In python i am using mlflow.login(). This requests me to enter a password. But i do not have any password due to the fact that databricks login only required a one time password.

Is there another way to connect python Code to the community Edition? Or how do i create a password?

In the settings in the community Edition there is a password changing functionality but this one Requires to enter the old password which obviously does not exist.

It would be great if someone could Provide some help! Thank you very much!

[Walter_C]

You can try the following:

Instead of using mlflow.login(), you can set the tracking URI and the token directly in your code. Here is an example:

import mlflow

# Set the tracking URI to your Databricks workspace
mlflow.set_tracking_uri("https://community.cloud.databricks.com")

# Set the Databricks token
token = "YOUR_PERSONAL_ACCESS_TOKEN"

# Configure the MLFlow client to use the token
mlflow.set_experiment("your-experiment-name")
mlflow.start_run()
mlflow.log_param("param1", 5)
mlflow.log_metric("metric1", 0.85)
mlflow.end_run()

[nicobuko]

@Walter_C Thank you for your fast answer 🙂

Where can I generate such a token in the community edition? I did not see an option to generate a token when I have looked into the community Edition.

[Walter_C]

Under Settings > Developer it does not allow you to?

[Rjdudley]

The Community Edition is very feature limited.  PATs are one feature which is not available.  None of the SQL Warehouse features are available, nor are most of the AI/ML features.

[Walter_C]

Got it, as per our docs it is suggested the creation of a token for Community edition as you can see here https://docs.databricks.com/en/mlflow/access-hosted-tracking-server.html#step-1-configure-your-envir... it states that for Community Edition you can use any of the two approaches listed there, can you go through the doc and let me know if this are the methods you have attempted?