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
googleapis · Nov 10, 2021 · eb8927f · eb8927f
1 parent 7f3b268
commit eb8927f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/samples/test/automlTablesPredict.v1beta1.test.js b/samples/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