ci: test docs notebooks too

Makefile

@@ -4,8 +4,8 @@ ONNXRUNTIME_VERSION := 1.15.1-?
 TOOLCHAIN_VERSION := 0.10.0-?
 LIBHAL_VERSION := 0.11.0-?
 
-.PHONY: check-docker-tag toolchain lint test unit_tests examples regression-test-all \
-regression-test-resnet50 regression-test-ssd-mobilenet \
+.PHONY: check-docker-tag toolchain lint test unit_tests notebook_tests examples \
+regression-test-all regression-test-resnet50 regression-test-ssd-mobilenet \
 regression-test-ssd-resnet34 regression-test-yolov5 doc docker-build docker-push \
 regression-test-efficientnet-b0 regression-test-efficientnet-v2-s
 
@@ -33,6 +33,9 @@ test:
 unit_tests:
 	pytest ./tests/unit/ -s
 
+notebook_tests:
+	pytest --nbmake ./docs
+
 examples:
 	for f in $$(find docs/examples/ -name *.py); do printf "\n[TEST] $$f ...\n"; python3 $$f || exit 1; done
 

docker/Dockerfile

@@ -14,8 +14,7 @@ RUN curl -fsSL https://cli.github.com/packages/githubcli-archive-keyring.gpg \
     | tee /etc/apt/sources.list.d/github-cli.list > /dev/null \
     && apt update && apt install gh -y
 
-RUN pip3 install --upgrade pip wheel setuptools Cython pytest pycocotools \
-    dvc[s3] pytest-benchmark
+RUN pip3 install --upgrade pip wheel setuptools Cython dvc[s3]
 
 RUN echo "deb [arch=amd64] https://internal-archive.furiosa.dev/ubuntu focal restricted" \
     > /etc/apt/sources.list.d/furiosa.list && \

docs/stylesheets/extra.css

@@ -1,4 +1,5 @@
 :root {
     --md-primary-fg-color: #780006;
+    --md-primary-fg-color--dark: #460101;
     --md-accent-fg-color: #e21500;
 }

pyproject.toml

@@ -47,6 +47,7 @@ test = [
     "pytest",
     "pytest-benchmark == 4.0.0",
     "pytest-asyncio ~= 0.17.2",
+    "nbmake",
     "pycocotools ~= 2.0.4",
 ]
 dvc = ["dvc[s3]"]