Fix estimation notebook tests (#8)

Author: jpn--

.github/workflows/core_tests.yml

@@ -332,31 +332,36 @@ jobs:
     steps:
       - uses: actions/checkout@v4
 
+      - name: "Set up Python"
+        uses: actions/setup-python@v5
+        with:
+          python-version-file: ".python-version"
+
       - name: Install uv
         uses: astral-sh/setup-uv@v5
         with:
           version: "0.7.12"
           enable-cache: true
           cache-dependency-glob: "uv.lock"
 
-      - name: "Set up Python"
-        uses: actions/setup-python@v5
-        with:
-          python-version-file: ".python-version"
+      - name: setup graphviz
+        uses: ts-graphviz/setup-graphviz@v2
 
       - name: Install activitysim
         run: |
-          uv sync --locked --only-group github-action
+          uv sync --locked --group github-action
 
       - name: Create Estimation Data
-        run: |
-          .venv\Scripts\python activitysim/examples/example_estimation/notebooks/est_mode_setup.py --household_sample_size 5000
+        run: >
+          uv run --group github-action python activitysim/examples/example_estimation/notebooks/est_mode_setup.py 
+          --household_sample_size 5000
 
       - name: Test Estimation Notebooks
-        run: |
-          uv run pytest activitysim/examples/example_estimation/notebooks/*.ipynb \
-            --nbmake-timeout=3000 \
-            --ignore=activitysim/examples/example_estimation/notebooks/01_estimation_mode.ipynb
+        run: >
+          uv run --group github-action pytest activitysim/examples/example_estimation/notebooks 
+          --nbmake-timeout=3000 
+          --ignore=activitysim/examples/example_estimation/notebooks/01_estimation_mode.ipynb 
+          --ignore-glob=activitysim/examples/example_estimation/notebooks/test-estimation-data/**
 
   estimation_edb_creation:
     needs: foundation
@@ -434,4 +439,4 @@ jobs:
           github_token: ${{ secrets.GITHUB_TOKEN }}
           # Token is created automatically by Github Actions, no other config needed
           publish_dir: ./docs/_build/html
-          destination_dir: develop
+          destination_dir: develop

activitysim/examples/example_estimation/build_full_mtc_example.py

@@ -30,6 +30,7 @@
 import os
 import shutil
 import subprocess
+import sys
 import time
 from pathlib import Path
 
@@ -74,7 +75,7 @@ def main(working_dir: Path, household_sample_size: int, skip_to_edb: bool = Fals
 
         subprocess.run(
             [
-                "python",
+                sys.executable,
                 "-m",
                 "activitysim",
                 "run",
@@ -115,7 +116,7 @@ def main(working_dir: Path, household_sample_size: int, skip_to_edb: bool = Fals
 
         subprocess.run(
             [
-                "python",
+                sys.executable,
                 str(infer_py),
                 str(pseudosurvey_dir),
                 str(configs_dir),
@@ -132,7 +133,7 @@ def main(working_dir: Path, household_sample_size: int, skip_to_edb: bool = Fals
 
     subprocess.run(
         [
-            "python",
+            sys.executable,
             "-m",
             "activitysim",
             "run",

pyproject.toml

@@ -157,11 +157,13 @@ github-action = [
     "dask==2023.11",
     "isort==5.12.0",
     "nbmake==1.4.6",
+    "pydot>=4.0.1",
     "pytest==7.2",
     "pytest-cov",
     "pytest-regressions",
     "requests>=2.32.3",
     "ruff",
+    "xlsxwriter==3.2.5",
     "zarr>=2,<3",
     "zstandard",
 ]