diff --git a/ai-platform/snippets/test/predict-image-classification.test.js b/ai-platform/snippets/test/predict-image-classification.test.js index ab196fc451..9711675eec 100644 --- a/ai-platform/snippets/test/predict-image-classification.test.js +++ b/ai-platform/snippets/test/predict-image-classification.test.js @@ -31,7 +31,8 @@ const endpointId = '71213169107795968'; const project = process.env.CAIP_PROJECT_ID; const location = 'us-central1'; -describe('AI platform predict image classification', () => { +describe('AI platform predict image classification', async function () { + this.retries(2); it('should make predictions using the image classification model', async () => { const stdout = execSync( `node ./predict-image-classification.js ${local_file} ${endpointId} ${project} ${location}`