From 59b4aab050498a3e57dcd042c57ba63697647d71 Mon Sep 17 00:00:00 2001
From: Karl Weinmeister <11586922+kweinmeister@users.noreply.github.com>
Date: Wed, 10 Nov 2021 14:10:10 -0600
Subject: [PATCH] test: Add retries to additional function in AutoML Tables
 Predict test (#564)

* Added retries to automlTablesPredict sample test

* Convert arrow functions

* Add retries to another function in predict test
---
 automl/test/automlTablesPredict.v1beta1.test.js | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/automl/test/automlTablesPredict.v1beta1.test.js b/automl/test/automlTablesPredict.v1beta1.test.js
index 92c65b702a5..77940a5a349 100644
--- a/automl/test/automlTablesPredict.v1beta1.test.js
+++ b/automl/test/automlTablesPredict.v1beta1.test.js
@@ -33,7 +33,10 @@ const bqOutputUriPrefix = `bq://${projectId}`;
 const exec = cmd => execSync(cmd, {encoding: 'utf8'});
 
 describe('Tables PredictionAPI', () => {
-  it('should perform single prediction', async () => {
+  it('should perform single prediction', async function () {
+    this.retries(5);
+    await delay(this.test);
+
     const inputs = [
       {numberValue: 39}, // Age
       {stringValue: 'technician'}, // Job