From 565fbefc00484c0c79a9e47daeb41624b21051d3 Mon Sep 17 00:00:00 2001
From: Raphael Mitsch <r.mitsch@outlook.com>
Date: Mon, 13 Nov 2023 14:52:08 +0100
Subject: [PATCH] Update spacy_llm/tests/models/test_hf.py

Co-authored-by: Sofie Van Landeghem <svlandeg@users.noreply.github.com>
---
 spacy_llm/tests/models/test_hf.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/spacy_llm/tests/models/test_hf.py b/spacy_llm/tests/models/test_hf.py
index 306e879a..3058035c 100644
--- 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": {