Skip to content

Commit e6e5f50

Browse files
committed
Fix bad test!
14 is correct (tested with python library)
1 parent 79a83d1 commit e6e5f50

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/models/jina_clip/test_processor_jina_clip.js

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,8 @@ export default () => {
3333
// Encode text and images
3434
const { input_ids, attention_mask, pixel_values } = await processor(sentences, images, { padding: true, truncation: true });
3535

36-
expect(input_ids.dims).toEqual([sentences.length, 19]);
37-
expect(attention_mask.dims).toEqual([sentences.length, 19]);
36+
expect(input_ids.dims).toEqual([sentences.length, 14]);
37+
expect(attention_mask.dims).toEqual([sentences.length, 14]);
3838
expect(pixel_values.dims).toEqual([images.length, 3, 512, 512]);
3939
expect(pixel_values.mean().item()).toBeCloseTo(0.7857685685157776, 6);
4040
},

0 commit comments

Comments
 (0)