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
GoogleCloudPlatform · Nov 14, 2022 · 6c570d3 · 6c570d3
1 parent af8d9d4
commit 6c570d3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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