Skip to content

Commit

Permalink
fix: typecheck
Browse files Browse the repository at this point in the history
Signed-off-by: axel7083 <[email protected]>
  • Loading branch information
axel7083 committed Jun 11, 2024
1 parent 4d9c621 commit 3caacd7
Showing 1 changed file with 5 additions and 6 deletions.
11 changes: 5 additions & 6 deletions packages/backend/src/workers/provider/InferenceProvider.spec.ts
Original file line number Diff line number Diff line change
Expand Up @@ -16,19 +16,20 @@
* SPDX-License-Identifier: Apache-2.0
***********************************************************************/

import { vi, describe, test, expect, beforeEach } from 'vitest';
import { beforeEach, describe, expect, test, vi } from 'vitest';
import type { TaskRegistry } from '../../registries/TaskRegistry';
import { type BetterContainerCreateResult, InferenceProvider } from './InferenceProvider';
import type { InferenceServerConfig } from '@shared/src/models/InferenceServerConfig';
import { containerEngine } from '@podman-desktop/api';
import type {
ContainerCreateOptions,
ContainerProviderConnection,
ImageInfo,
ProviderContainerConnection,
ContainerCreateOptions,
} from '@podman-desktop/api';
import { containerEngine } from '@podman-desktop/api';
import { getImageInfo, getProviderContainerConnection } from '../../utils/inferenceUtils';
import type { TaskState } from '@shared/src/models/ITask';
import { InferenceType } from '@shared/src/models/IInference';

vi.mock('../../utils/inferenceUtils', () => ({
getProviderContainerConnection: vi.fn(),
Expand Down Expand Up @@ -61,10 +62,8 @@ const taskRegistry: TaskRegistry = {
} as unknown as TaskRegistry;

class TestInferenceProvider extends InferenceProvider {
name: string = 'test-inference-provider';

constructor() {
super(taskRegistry);
super(taskRegistry, InferenceType.NONE, 'test-inference-provider');
}

enabled(): boolean {
Expand Down

0 comments on commit 3caacd7

Please sign in to comment.