Merge branch 'main' into broken-link-scanner

pymc-labs · Nov 5, 2024 · 461a892 · 461a892
2 parents 5870345 + eb80ddb
commit 461a892
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 7 deletions.
diff --git a/.github/workflows/slow-tests-issue.yml b/.github/workflows/slow-tests-issue.yml
@@ -13,6 +13,8 @@ jobs:
   update-comment:
     runs-on: ubuntu-latest
     steps:
+      - name: Install ZSH
+        run: sudo apt-get update && sudo apt-get install -y zsh
       - name: Checkout code
         uses: actions/checkout@v2
       - name: Set up Python
@@ -21,6 +23,7 @@ jobs:
           python-version: "3.11"
       - name: Trigger the script
         working-directory: scripts/slowest_tests
+        shell: zsh {0}
         run: source update-slowest-times-issue.sh
         env:
           GITHUB_TOKEN: ${{ github.token }}
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()"
    ]
   },
   {

diff --git a/scripts/slowest_tests/update-slowest-times-issue.sh b/scripts/slowest_tests/update-slowest-times-issue.sh
@@ -1,4 +1,4 @@
-#!/bin/bash
+#!/bin/zsh
 
 DRY_RUN=false