Skip to content

Commit

Permalink
remove T4, A100, TPU profil options for jupyterhub
Browse files Browse the repository at this point in the history
  • Loading branch information
chiayi committed Apr 1, 2024
1 parent 0c7c34e commit 01d2bc0
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion cloudbuild.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -249,8 +249,14 @@ steps:
cd /workspace/applications/rag/tests
python3 test_frontend.py "127.0.0.1:8081"
echo "pass" > /workspace/rag_frontend_result.txt
sleep 5s
cd /workspace/
find . -type f -name "*.ipynb" > notebook_file_list.txt
while IFS= read -r line; do gsutil cp $line gs://gke-aieco-rag-$SHORT_SHA-$_BUILD_ID/notebooks/ ; done < notebook_file_list.txt
ray job submit --working-dir . --address=http://127.0.0.1:8265 -- jupyter nbconvert --to notebook --execute /notebooks/rag-kaggle-ray-sql-latest.ipynb
allowFailure: true
waitFor: ['cleanup jupyterhub', 'cleanup ray cluster']
waitFor: ['cleanup jupyterhub', 'cleanup ray cluster']

- id: 'cleanup rag'
name: 'gcr.io/$PROJECT_ID/terraform'
Expand Down

0 comments on commit 01d2bc0

Please sign in to comment.