@@ -128,19 +128,28 @@ jobs:
128128 cloud_runner : linux-x86-n2-16-buildkit
129129
130130 maxtext_jupyter_notebooks :
131+ name : Run Jupyter notebook examples
131132 needs : build_and_upload_maxtext_package
132- if : needs.analyze_code_changes.outputs.run_notebooks == 'true'
133- uses : ./.github/workflows/run_jupyter_notebooks.yml
134- strategy :
135- fail-fast : false
136- with :
137- device_type : tpu
138- device_name : v6e-4
139- base_image : maxtext-unit-test-tpu:py312
140- cloud_runner : linux-x86-ct6e-180-4tpu
141- maxtext_sha : ${{ needs.build_and_upload_maxtext_package.outputs.maxtext_sha }}
142- secrets :
143- HF_TOKEN : ${{ secrets.HF_TOKEN }}
133+ # if: needs.analyze_code_changes.outputs.run_notebooks == 'true'
134+ # uses: ./.github/workflows/run_jupyter_notebooks.yml
135+ steps :
136+ - name : Echo variable
137+ run : echo "Echo MY_HF_TOKEN $MY_HF_TOKEN"
138+ env :
139+ MY_HF_TOKEN : ${{ secrets.HF_TOKEN }}
140+ - name : Run Jupyter Notebooks
141+ if : needs.analyze_code_changes.outputs.run_notebooks == 'true'
142+ uses : ./.github/workflows/run_jupyter_notebooks.yml
143+ strategy :
144+ fail-fast : false
145+ with :
146+ device_type : tpu
147+ device_name : v6e-4
148+ base_image : maxtext-unit-test-tpu:py312
149+ cloud_runner : linux-x86-ct6e-180-4tpu
150+ maxtext_sha : ${{ needs.build_and_upload_maxtext_package.outputs.maxtext_sha }}`
151+ secrets :
152+ HF_TOKEN : ${{ secrets.HF_TOKEN }}
144153
145154 tpu-tests :
146155 name : ${{ matrix.flavor }} tests
@@ -301,4 +310,4 @@ jobs:
301310 if : ${{ contains(needs.*.result, 'failure') && github.event_name == 'schedule' }}
302311 uses : jayqi/failed-build-issue-action@1a893bbf43ef1c2a8705e2b115cd4f0fe3c5649b # v1.2.0
303312 with :
304- github-token : ${{ secrets.GITHUB_TOKEN }}
313+ github-token : ${{ secrets.GITHUB_TOKEN }}
0 commit comments