Merge pull request #31 from databricks-industry-solutions/fix-mlflow-uc

set-registry-to-uc
databricks-industry-solutions · Aug 6, 2024 · cc436b9 · cc436b9
2 parents f5a906c + cde2d59
commit cc436b9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/01_causal_discovery.ipynb b/01_causal_discovery.ipynb
@@ -491,7 +491,7 @@
    "language": "python",
    "notebookMetadata": {
     "mostRecentlyExecutedCommandWithImplicitDF": {
-     "commandId": 888983575928260,
+     "commandId": 2548317818328688,
      "dataframes": [
       "_sqldf"
      ]

diff --git a/util/notebook-config.py b/util/notebook-config.py
@@ -17,6 +17,8 @@
 %matplotlib inline
 %config InlineBackend.figure_format = "retina"
 
+mlflow.set_registry_uri("databricks-uc")
+
 # COMMAND ----------
 
 # Create catalog if it doesn't exist