upload rendered notebook to gist

.github/workflows/nsys-jax.yaml

@@ -72,7 +72,7 @@ jobs:
           ./venv/bin/mypy ${NSYS_JAX_PYTHON_FILES}
 
   notebook:
-    runs-on: ubuntu-22.04
+    runs-on: ubuntu-24.04
     steps:
       - name: Check out the repository under ${GITHUB_WORKSPACE}
         uses: actions/checkout@v4
@@ -95,7 +95,48 @@ jobs:
         run: |
           pushd .github/container/jax_nsys
           ./nsys_jax_venv/bin/python -m jupyterlab --version
+          # Run with ipython for the sake of getting a clear error message
           ./nsys_jax_venv/bin/ipython Analysis.ipynb
+      - name: Render the notebook
+        id: render
+        shell: bash -x -e {0}
+        run: |
+          pushd .github/container/jax_nsys
+          workdir=$(mktemp -d)
+          ./nsys_jax_venv/bin/jupyter nbconvert --execute --to notebook --output-dir=$workdir Analysis.ipynb
+          echo "WORKDIR=$workdir" >> $GITHUB_OUTPUT
+      - name: Upload rendered notebook to gist
+        id: upload
+        uses: actions/github-script@v7
+        with:
+          github-token: ${{ secrets.NVJAX_GIST_TOKEN }}
+          script: |
+            const currentDateTime = new Date().toISOString();
+            const gistDescription =
+              `Rendered IPython notebook from workflow: ${{ github.workflow }}, ` +
+              `Run ID: ${{ github.run_id }}, ` +
+              `Repository: ${{ github.repository }}, ` +
+              `Event: ${{ github.event_name }}, ` +
+              `Created: ${currentDateTime}`;
+            const fs = require('fs').promises;
+            const workdir = '${{ steps.render.outputs.WORKDIR }}'
+            const files = await fs.readdir(workdir);
+            gist = await github.rest.gists.create({
+              description: gistDescription,
+              public: false,
+              files: Object.fromEntries(
+                await Promise.all(
+                  files.map(
+                    async filename => {
+                      const content = await fs.readFile(`${workdir}/${filename}`, 'utf8');
+                      return [filename, { content }];
+                    }
+                  )
+                )
+              )
+            });
+            console.log(gist)
+            return gist.data.id;
 
   ruff:
     runs-on: ubuntu-24.04