Skip to content

Commit b4f287f

Browse files
committed
Increase timeout for batch prediciton
1 parent 1d05bb8 commit b4f287f

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

ai-platform/snippets/create-hyperparameter-tuning-job.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ function main(
7676
{
7777
machineSpec: {
7878
machineType: 'n1-standard-4',
79-
acceleratorType: 'NVIDIA_TESLA_K80',
79+
acceleratorType: 'NVIDIA_TESLA_T4',
8080
acceleratorCount: 1,
8181
},
8282
replicaCount: 1,

ai-platform/snippets/test/batch-prediction-gemini.test.js

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ describe('Batch predict with Gemini', async () => {
7474
batchPredictionGcsJobId = response
7575
.split('/locations/us-central1/batchPredictionJobs/')[1]
7676
.split('\n')[0];
77-
}).timeout(10000);
77+
}).timeout(100000);
7878

7979
it('should create Batch prediction Gemini job with BigQuery', async () => {
8080
const response = execSync(
@@ -85,5 +85,5 @@ describe('Batch predict with Gemini', async () => {
8585
batchPredictionBqJobId = response
8686
.split('/locations/us-central1/batchPredictionJobs/')[1]
8787
.split('\n')[0];
88-
}).timeout(10000);
88+
}).timeout(100000);
8989
});

0 commit comments

Comments
 (0)