Update spacy_llm/tests/models/test_hf.py

Co-authored-by: Sofie Van Landeghem <[email protected]>
explosion · Nov 13, 2023 · 565fbef · 565fbef
1 parent d6e4c24
commit 565fbef
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/spacy_llm/tests/models/test_hf.py b/spacy_llm/tests/models/test_hf.py
@@ -4,9 +4,7 @@
 import spacy
 from thinc.compat import has_torch_cuda_gpu
 
-from spacy_llm.compat import has_accelerate
-
-from ...compat import torch
+from spacy_llm.compat import has_accelerate, torch
 
 _PIPE_CFG = {
     "model": {