From 31f156a1f998504e4081b699fdc3861ea3bcf7a0 Mon Sep 17 00:00:00 2001
From: nayaknishant <nishantnayak@google.com>
Date: Tue, 10 May 2022 15:26:31 -0700
Subject: [PATCH] fix: added retries to flaky test (#299)

---
 ai-platform/snippets/test/create-custom-job.test.js | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ai-platform/snippets/test/create-custom-job.test.js b/ai-platform/snippets/test/create-custom-job.test.js
index 5bd73f90d6..ac87d2e9bd 100644
--- a/ai-platform/snippets/test/create-custom-job.test.js
+++ b/ai-platform/snippets/test/create-custom-job.test.js
@@ -43,7 +43,8 @@ function parseResponse(stdout) {
 
 let customJobId;
 
-describe('AI platform create custom job', () => {
+describe('AI platform create custom job', async function () {
+  this.retries(2);
   it('should create a new custom job', async () => {
     const stdout = execSync(
       `node ./create-custom-job.js ${customJobDisplayName} \