Update MMM docs to use graphviz method (#1172)

* Update docs to use graphviz method * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: Will Dean <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
pymc-labs · Nov 5, 2024 · eb80ddb · eb80ddb
1 parent bd5261c
commit eb80ddb
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git 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",
@@ -6422,7 +6421,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
@@ -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
@@ -707,7 +707,7 @@
    ],
    "source": [
     "# Model to graphiz\n",
-    "pm.model_to_graphviz(dummy_mmm.model)"
+    "dummy_mmm.graphviz()"
    ]
   },
   {