From cde2d59cb5ab76dca6342b3d1d4c4687af22c3ca Mon Sep 17 00:00:00 2001
From: ryuta-yoshimatsu <ryuta-yoshimatsu>
Date: Tue, 6 Aug 2024 08:33:39 +0000
Subject: [PATCH] set-registry-to-uc

---
 01_causal_discovery.ipynb | 2 +-
 util/notebook-config.py   | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/01_causal_discovery.ipynb b/01_causal_discovery.ipynb
index ec47945..d26643c 100644
--- 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
index cb8f2a3..0519068 100644
--- 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