Skip to content

Commit

Permalink
test: Add retries to additional function in AutoML Tables Predict test (
Browse files Browse the repository at this point in the history
#564)

* Added retries to automlTablesPredict sample test

* Convert arrow functions

* Add retries to another function in predict test
  • Loading branch information
kweinmeister authored Nov 10, 2021
1 parent 09eab1c commit 23d9f1f
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion automl/test/automlTablesPredict.v1beta1.test.js
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 23d9f1f

Please sign in to comment.