From cb97b0bc05cc3f9d5881b11e8f55db0732456373 Mon Sep 17 00:00:00 2001
From: giom <giom@google.com>
Date: Wed, 1 Jan 2025 12:48:49 +0000
Subject: [PATCH] Formatting notebooks

---
 .github/workflows/notebooks.yaml | 34 ++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/.github/workflows/notebooks.yaml b/.github/workflows/notebooks.yaml
index 4065f5cf7..3c7f96e90 100644
--- a/.github/workflows/notebooks.yaml
+++ b/.github/workflows/notebooks.yaml
@@ -24,6 +24,40 @@ jobs:
       run: python3 -m pip install -U git+https://github.com/tensorflow/docs
     - name: Fetch main branch
       run: git fetch -u origin main:main
+    - name: Get changed notebooks
+      id: changed-notebooks
+      run: |
+        if [ "${{ github.event_name }}" == "pull_request" ]; then
+          # Only check notebooks modified in this pull request
+          git diff --name-only origin/main | grep '\.ipynb$' > changed_notebooks.txt
+        else
+          # Manual run, check everything
+          find . -name '*.ipynb' > changed_notebooks.txt
+        fi
+        echo "changed_notebooks=$(cat changed_notebooks.txt)" >> $GITHUB_OUTPUT
+    - name: Format notebooks
+      run: |
+        if [ -s "${{ steps.changed-notebooks.outputs.changed_notebooks }}" ]; then
+          echo "Formatting notebooks with nbfmt:"
+          python3 -m tensorflow_docs.tools.nbfmt ${{ steps.changed-notebooks.outputs.changed_notebooks }}
+        else
+          echo "No notebooks to format."
+        fi
+    - name: Commit and push changes (if any)
+      if: ${{ success() }}
+      run: |
+        git config --local user.email "action@github.com"
+        git config --local user.name "GitHub Action"
+        # Check if there are changes
+        if ! git diff --quiet; then
+          # Commit on the current branch
+          git add .
+          git commit -m "Format notebooks"
+          # Push directly to the PR branch
+          git push origin ${{ github.event.pull_request.head.ref }}
+        else
+          echo "No changes to commit."
+        fi
     - name: Check notebook formatting
       run: |
         if [ "${{ github.event_name }}" == "pull_request" ]; then