Skip to content

Commit f92a3ee

Browse files
committed
Fixed ci and lint error
1 parent 1f4fe27 commit f92a3ee

File tree

5 files changed

+15
-34
lines changed

5 files changed

+15
-34
lines changed

.github/workflows/ci.yml

Lines changed: 0 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -49,16 +49,3 @@ jobs:
4949

5050
- name: Build package
5151
run: yarn prepare
52-
53-
build-web:
54-
runs-on: ubuntu-latest
55-
steps:
56-
- name: Checkout
57-
uses: actions/checkout@v4
58-
59-
- name: Setup
60-
uses: ./.github/actions/setup
61-
62-
- name: Build example for Web
63-
run: |
64-
yarn example expo export --platform web

src/__tests__/text-embedding.pipeline.test.tsx

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -11,12 +11,6 @@ jest.mock('../models/text-embedding', () => {
1111
};
1212
});
1313

14-
// Create a callable tokenizer mock
15-
const createCallableTokenizer = () => {
16-
const tokenizer = jest.fn().mockResolvedValue({ input_ids: [1n, 2n, 3n] });
17-
return tokenizer;
18-
};
19-
2014
describe('TextEmbedding Pipeline', () => {
2115
beforeEach(() => {
2216
jest.clearAllMocks();

src/pipelines/text-embedding.tsx

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,4 @@
1-
import {
2-
env,
3-
AutoTokenizer,
4-
} from '@huggingface/transformers';
1+
import { env, AutoTokenizer } from '@huggingface/transformers';
52
import type { PreTrainedTokenizer } from '@huggingface/transformers';
63
import { TextEmbedding as Model } from '../models/text-embedding';
74
import type { LoadOptions } from '../models/base';

src/pipelines/text-generation.tsx

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,4 @@
1-
import {
2-
env,
3-
AutoTokenizer,
4-
} from '@huggingface/transformers';
1+
import { env, AutoTokenizer } from '@huggingface/transformers';
52
import type { PreTrainedTokenizer } from '@huggingface/transformers';
63
import { TextGeneration as Model } from '../models/text-generation';
74
import type { LoadOptions } from '../models/base';

src/types/huggingface-transformers.d.ts

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,18 @@
11
declare module '@huggingface/transformers' {
22
export interface PreTrainedTokenizer {
3-
(text: string, options?: {
4-
return_tensor?: boolean;
5-
padding?: boolean;
6-
truncation?: boolean;
7-
max_length?: number;
8-
}): Promise<{ input_ids: number[] }>;
9-
decode(tokens: number[], options?: { skip_special_tokens?: boolean }): string;
3+
(
4+
text: string,
5+
options?: {
6+
return_tensor?: boolean;
7+
padding?: boolean;
8+
truncation?: boolean;
9+
max_length?: number;
10+
}
11+
): Promise<{ input_ids: number[] }>;
12+
decode(
13+
tokens: number[],
14+
options?: { skip_special_tokens?: boolean }
15+
): string;
1016
}
1117

1218
export class AutoTokenizer {

0 commit comments

Comments
 (0)