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 17, 2022 · 5801dd0 · 5801dd0
1 parent 188bced
commit 5801dd0
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