Add pytype/pylint testing of notebooks

PiperOrigin-RevId: 590572464 Change-Id: If7331ae624c99c36ec9c32de7b4b12ce7edfe561
google-deepmind · Dec 13, 2023 · 4218be2 · 4218be2
1 parent 1481dad
commit 4218be2
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/.github/actions/install-examples/action.yml b/.github/actions/install-examples/action.yml
@@ -13,3 +13,10 @@ runs:
     - name: Show installed dependencies
       shell: bash
       run: pip list
+
+    - name: Convert notebooks
+      run: |
+        jupyter nbconvert --to=python --output-dir=notebooks \
+            --RegexRemovePreprocessor.patterns='!pip' \
+            $(find examples -type f -iname '*.ipynb')
+        ls notebooks/*.py
diff --git a/.github/workflows/pylint-examples.yml b/.github/workflows/pylint-examples.yml
@@ -45,3 +45,6 @@ jobs:
 
       - name: Run PyLint on examples
         run: pylint --errors-only examples
+
+      - name: Run PyLint on notebooks
+        run: pylint --errors-only notebooks
diff --git a/.github/workflows/pytype-examples.yml b/.github/workflows/pytype-examples.yml
@@ -45,3 +45,6 @@ jobs:
 
       - name: Run PyType on examples
         run: pytype examples
+
+      - name: Run PyType on notebooks
+        run: pytype --pythonpath=. notebooks