From f36d0f62bd45821f15e3ade6b654fa3cefe42cdb Mon Sep 17 00:00:00 2001
From: takechanman1228 <takeda.hajime.ja@gmail.com>
Date: Tue, 5 Nov 2024 11:50:30 -0500
Subject: [PATCH 1/2] Update docs to use graphviz method

---
 docs/source/notebooks/mmm/mmm_example.ipynb                 | 2 +-
 docs/source/notebooks/mmm/mmm_lift_test.ipynb               | 6 +++---
 .../notebooks/mmm/mmm_time_varying_media_example.ipynb      | 2 +-
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/docs/source/notebooks/mmm/mmm_example.ipynb b/docs/source/notebooks/mmm/mmm_example.ipynb
index 3d32a893d..d1d71278e 100644
--- a/docs/source/notebooks/mmm/mmm_example.ipynb
+++ b/docs/source/notebooks/mmm/mmm_example.ipynb
@@ -6422,7 +6422,7 @@
     }
    ],
    "source": [
-    "pm.model_to_graphviz(model=mmm.model)"
+    "mmm.graphviz()"
    ]
   },
   {
diff --git a/docs/source/notebooks/mmm/mmm_lift_test.ipynb b/docs/source/notebooks/mmm/mmm_lift_test.ipynb
index cb8f69729..84e815e70 100644
--- a/docs/source/notebooks/mmm/mmm_lift_test.ipynb
+++ b/docs/source/notebooks/mmm/mmm_lift_test.ipynb
@@ -1211,7 +1211,7 @@
     }
    ],
    "source": [
-    "pm.model_to_graphviz(mmm.model)"
+    "mmm.graphviz()"
    ]
   },
   {
@@ -1430,7 +1430,7 @@
    "source": [
     "mmm.add_lift_test_measurements(df_lift_test)\n",
     "\n",
-    "pm.model_to_graphviz(mmm.model)"
+    "mmm.graphviz()"
    ]
   },
   {
@@ -2077,7 +2077,7 @@
    "source": [
     "mmm.add_lift_test_measurements(df_additional_lift_test, name=\"more_lift_measurements\")\n",
     "\n",
-    "pm.model_to_graphviz(mmm.model)"
+    "mmm.graphviz()"
    ]
   },
   {
diff --git a/docs/source/notebooks/mmm/mmm_time_varying_media_example.ipynb b/docs/source/notebooks/mmm/mmm_time_varying_media_example.ipynb
index 8ab206d58..307ebe0f9 100644
--- a/docs/source/notebooks/mmm/mmm_time_varying_media_example.ipynb
+++ b/docs/source/notebooks/mmm/mmm_time_varying_media_example.ipynb
@@ -707,7 +707,7 @@
    ],
    "source": [
     "# Model to graphiz\n",
-    "pm.model_to_graphviz(dummy_mmm.model)"
+    "dummy_mmm.graphviz()"
    ]
   },
   {

From 6bccc5ca2e562c410ca6561624185c05ef28b30d Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Tue, 5 Nov 2024 17:22:54 +0000
Subject: [PATCH 2/2] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 docs/source/notebooks/mmm/mmm_example.ipynb | 1 -
 1 file changed, 1 deletion(-)

diff --git a/docs/source/notebooks/mmm/mmm_example.ipynb b/docs/source/notebooks/mmm/mmm_example.ipynb
index d1d71278e..16df5f536 100644
--- a/docs/source/notebooks/mmm/mmm_example.ipynb
+++ b/docs/source/notebooks/mmm/mmm_example.ipynb
@@ -187,7 +187,6 @@
     "import matplotlib.pyplot as plt\n",
     "import numpy as np\n",
     "import pandas as pd\n",
-    "import pymc as pm\n",
     "import seaborn as sns\n",
     "\n",
     "from pymc_marketing.mmm import MMM, GeometricAdstock, LogisticSaturation\n",