Use local hub library for unit tests

huggingface · Nov 22, 2023 · 7cb9849 · 7cb9849
1 parent 246514e
commit 7cb9849
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/packages/templates/test/e2e.test.js b/packages/templates/test/e2e.test.js
@@ -1,9 +1,8 @@
-import { assert, describe, expect, it } from 'vitest';
+import { describe, expect, it } from 'vitest';
 
 import { Template } from '../src/index.js';
 
-import { downloadFile } from '@huggingface/hub';
-
+import { downloadFile } from '../../hub';
 
 const EXAMPLE_CHAT = [
     { "role": "user", "content": "Hello, how are you?" },