Skip to content

Commit d6c24b6

Browse files
sofislpattishin
authored andcommitted
build: increase retries and timeout for flaky tests (#60)
* build: increase retries and timeout for flaky tests
1 parent 4014d50 commit d6c24b6

11 files changed

+21
-11
lines changed

ai-platform/snippets/package.json

+1-1
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
"*.js"
1111
],
1212
"scripts": {
13-
"test": "mocha --timeout 600000 test/*.js"
13+
"test": "mocha --timeout 800000 test/*.js"
1414
},
1515
"dependencies": {
1616
"@google-cloud/aiplatform": "^1.3.0"

ai-platform/snippets/test/create-training-pipeline-image-classification.test.js

+2-1
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,8 @@ const project = process.env.CAIP_PROJECT_ID;
4040

4141
let trainingPipelineId;
4242

43-
describe('AI platform create training pipeline image classification', () => {
43+
describe('AI platform create training pipeline image classification', async function () {
44+
this.retries(2);
4445
it('should create a new image classification training pipeline', async () => {
4546
const stdout = execSync(
4647
`node ./create-training-pipeline-image-classification.js ${datasetId} ${modelDisplayName} ${trainingPipelineDisplayName} ${project} ${location}`

ai-platform/snippets/test/create-training-pipeline-image-object-detection.test.js

+2-1
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,8 @@ const project = process.env.CAIP_PROJECT_ID;
4242

4343
let trainingPipelineId;
4444

45-
describe('AI platform create training pipeline image object detection', () => {
45+
describe('AI platform create training pipeline image object detection', async function () {
46+
this.retries(2);
4647
it('should create a new image object detection training pipeline', async () => {
4748
const stdout = execSync(
4849
`node ./create-training-pipeline-image-object-detection.js \

ai-platform/snippets/test/create-training-pipeline-tabular-classification.test.js

+2-1
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,8 @@ const project = process.env.CAIP_PROJECT_ID;
4343

4444
let trainingPipelineId;
4545

46-
describe('AI platform create training pipeline tables classification', () => {
46+
describe('AI platform create training pipeline tables classification', async function () {
47+
this.retries(2);
4748
it('should create a new tables classification training pipeline', async () => {
4849
const stdout = execSync(
4950
`node ./create-training-pipeline-tabular-classification.js \

ai-platform/snippets/test/create-training-pipeline-tabular-regression.test.js

+2-1
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,8 @@ const project = process.env.CAIP_PROJECT_ID;
4343

4444
let trainingPipelineId;
4545

46-
describe('AI platform create training pipeline tabular regression', () => {
46+
describe('AI platform create training pipeline tabular regression', async function () {
47+
this.retries(2);
4748
it('should create a new tabular regression training pipeline', async () => {
4849
const stdout = execSync(
4950
`node ./create-training-pipeline-tabular-regression.js \

ai-platform/snippets/test/create-training-pipeline-text-classification.test.js

+2-1
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,8 @@ const project = process.env.CAIP_PROJECT_ID;
4242

4343
let trainingPipelineId;
4444

45-
describe('AI platform create training pipeline text classification', () => {
45+
describe('AI platform create training pipeline text classification', async function () {
46+
this.retries(2);
4647
it('should create a new text classification training pipeline', async () => {
4748
const stdout = execSync(
4849
`node ./create-training-pipeline-text-classification.js \

ai-platform/snippets/test/create-training-pipeline-text-entity-extraction.test.js

+2-1
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,8 @@ const project = process.env.CAIP_PROJECT_ID;
4242

4343
let trainingPipelineId;
4444

45-
describe('AI platform create training pipeline text entity extraction', () => {
45+
describe('AI platform create training pipeline text entity extraction', async function () {
46+
this.retries(2);
4647
it('should create a new text entity extraction training pipeline', async () => {
4748
const stdout = execSync(
4849
`node ./create-training-pipeline-text-entity-extraction.js \

ai-platform/snippets/test/create-training-pipeline-text-sentiment-analysis.test.js

+2-1
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,8 @@ const project = process.env.CAIP_PROJECT_ID;
4242

4343
let trainingPipelineId;
4444

45-
describe('AI platform create training pipeline text sentiment analysis', () => {
45+
describe('AI platform create training pipeline text sentiment analysis', async function () {
46+
this.retries(2);
4647
it('should create a new text sentiment analysis training pipeline', async () => {
4748
const stdout = execSync(
4849
`node ./create-training-pipeline-text-sentiment-analysis.js \

ai-platform/snippets/test/create-training-pipeline-video-classification.test.js

+2-1
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,8 @@ const project = process.env.CAIP_PROJECT_ID;
4242

4343
let trainingPipelineId;
4444

45-
describe('AI platform create training pipeline video classification', () => {
45+
describe('AI platform create training pipeline video classification', async function () {
46+
this.retries(2);
4647
it('should create a new video classification training pipeline', async () => {
4748
const stdout = execSync(
4849
`node ./create-training-pipeline-video-classification.js ${datasetId} \

ai-platform/snippets/test/create-training-pipeline-video-object-tracking.test.js

+2-1
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,8 @@ const project = process.env.CAIP_PROJECT_ID;
4242

4343
let trainingPipelineId;
4444

45-
describe('AI platform create training pipeline object tracking', () => {
45+
describe('AI platform create training pipeline object tracking', async function () {
46+
this.retries(2);
4647
it('should create a new object tracking training pipeline', async () => {
4748
const stdout = execSync(
4849
`node ./create-training-pipeline-video-object-tracking.js \

ai-platform/snippets/test/predict-image-object-detection.test.js

+2-1
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,8 @@ const endpointId = '2791387344039575552';
2929
const project = process.env.CAIP_PROJECT_ID;
3030
const location = 'us-central1';
3131

32-
describe('AI platform predict image object detection', () => {
32+
describe('AI platform predict image object detection', async function () {
33+
this.retries(2);
3334
it('should make predictions using the image object detection model', async () => {
3435
const stdout = execSync(
3536
`node ./predict-image-object-detection.js ${filename} \

0 commit comments

Comments
 (0)