Options
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
06-18-2021 01:55 PM
You could monkey patch MLFlow's _get_dbutils() with something similar to this to get this working while connecting from dbconnect
spark = SparkSession.builder.getOrCreate()
# monkey-patch MLFlow's _get_dbutils()
def _get_dbutils():
return DBUtils(spark)
mlflow.utils.databricks_utils._get_dbutils = _get_dbutils