Skip to content

Commit 43976bb

Browse files
committed
chore: wip
1 parent 2bf6a71 commit 43976bb

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

storage/framework/core/ai/tests/ai.test.ts

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -180,10 +180,18 @@ describe('Bedrock Client', () => {
180180
trainingDataConfig: {
181181
s3Uri: 's3://example-bucket/training-data',
182182
},
183+
validationDataConfig: {
184+
s3Uri: 's3://example-bucket/validation-data',
185+
},
183186
outputDataConfig: {
184187
s3Uri: 's3://example-bucket/output-data',
185188
},
186189
hyperParameters: {},
190+
trainingMetrics: {
191+
trainingLoss: 0.1,
192+
validationLoss: 0.2,
193+
epochsCompleted: 10,
194+
},
187195
}
188196

189197
bedrockMock.on(GetModelCustomizationJobCommand).resolves(mockResponse)

0 commit comments

Comments
 (0)