Not able to run end to end ML project on Databricks Trial

Sri2025
New Contributor

I started using Databricks trial version from today. I want to explore full end to end ML lifecycle on the databricks. I observed for the compute only 'serverless' option is available. I was trying to execute the notebook posted on https://docs.databricks.com/aws/en/mlflow/end-to-end-example. However, I am getting exception in the stage related to 'model registration' (

model_version = mlflow.register_model(f"runs:/{run_id}/random_forest_model", model_name)).
 
It is giving below error and the model is not getting registered -
 
ConnectionClosedError: Connection was closed before we received a valid response from endpoint URL: "https://s3.us-west-2.amazonaws.com/dbstorage-prod-ro8nd".
 
Can you please advise.