reduce tabicl train size in CI

.github/workflows/deploy_book.yml

@@ -10,6 +10,8 @@ jobs:
     permissions:
       contents: write
     runs-on: ubuntu-latest
+    env:
+      TABICL_TRAIN_SIZE: "1500"
     steps:
       - uses: actions/checkout@v4
         with:

.github/workflows/testing.yml

@@ -14,6 +14,8 @@ jobs:
       matrix:
         os: [windows-latest, ubuntu-latest, macos-latest]
     runs-on: ${{ matrix.os }}
+    env:
+      TABICL_TRAIN_SIZE: "1500"
     steps:
       - uses: actions/checkout@v4
       - uses: prefix-dev/setup-pixi@v0.10.0

.gitignore

@@ -85,6 +85,7 @@ content/results
 book/_build
 content/notebooks
 content/python_files/*_lib.py
+content/python_files/*.ipynb
 
 # jupyterlite build
 jupyterlite/.jupyterlite.doit.db

content/python_files/prediction_intervals_tabicl.py

@@ -46,7 +46,11 @@ def tabicl_quantiles_to_df(prediction, quantiles, mode=skrub.eval_mode()):
     return pl.DataFrame(prediction, schema=[f"q_{q}" for q in quantiles])
 
 
-def limit_train_size(df, size=9000, mode=skrub.eval_mode()):
+def limit_train_size(
+    df,
+    size=tutorial_helpers.TABICL_TRAIN_SIZE,  # 9000 by default, 1500 in CI
+    mode=skrub.eval_mode(),
+):
     if mode in ("fit", "fit_transform", "preview"):
         return df.tail(size)
     return df

content/python_files/tutorial_helpers.py

@@ -1,4 +1,5 @@
 import datetime
+import os
 
 import altair
 import numpy as np
@@ -7,6 +8,8 @@
 import polars.selectors as cs
 import skrub
 
+TABICL_TRAIN_SIZE = int(os.environ.get("TABICL_TRAIN_SIZE", "9000"))
+
 
 def lorenz_curve(observed_value, predicted_value, n_samples=1_000):
     """Compute the Lorenz curve for a given true and predicted values.

pixi.toml

@@ -11,7 +11,7 @@ create-notebooks-dir = { cmd = "mkdir -p ./content/notebooks" }
 scrape-function-definitions = { cmd = "python utils/extract_defs.py" }
 copy-tutorial-helpers = { cmd = "cp ./content/python_files/tutorial_helpers.py ./content/notebooks/tutorial_helpers.py && cp ./content/python_files/*_lib.py ./content/notebooks/", depends-on = ["create-notebooks-dir", "scrape-function-definitions"] }
 convert-to-notebooks = { cmd = "python utils/convert_to_notebooks.py", depends-on = ["create-notebooks-dir", "copy-tutorial-helpers", "scrape-function-definitions"] }
-convert-to-executed-notebooks = { cmd = "python utils/convert_to_notebooks.py --execute --exclude single_horizon_prediction.py --exclude tabicl_time_series_forecasting.py", depends-on = ["create-notebooks-dir", "copy-tutorial-helpers", "scrape-function-definitions"] }
+convert-to-executed-notebooks = { cmd = "python utils/convert_to_notebooks.py --execute", depends-on = ["create-notebooks-dir", "copy-tutorial-helpers", "scrape-function-definitions"] }
 build-book = { cmd = "jupyter-book build book", depends-on = ["convert-to-notebooks", "copy-tutorial-helpers", "scrape-function-definitions"] }
 build-jupyterlite = { cmd = "jupyter lite build --contents content --output-dir dist", cwd = "jupyterlite", depends-on = ["convert-to-notebooks"] }
 serve-jupyterlite = { cmd = "python -m http.server", cwd = "jupyterlite/dist", depends-on = ["build-jupyterlite"] }
@@ -49,6 +49,7 @@ jupyter-book = ">=1.0.4,<2"
 jupyterlite = "*"
 jupyterlite-pyodide-kernel = "*"
 python-libarchive-c = "*"
+tabicl = ">=2.2.0"
 
 [pypi-dependencies]
 openmeteo-requests = ">=1.5.0, <2"