From a24fd05fe7376d263ca00bd7c192e86a45c5249e Mon Sep 17 00:00:00 2001
From: MatteoRobbiati <robbiati.matteo@gmail.com>
Date: Tue, 19 Nov 2024 09:15:56 +0100
Subject: [PATCH] fix: importing pytorch backend from qiboml in the tests too

---
 tests/conftest.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/tests/conftest.py b/tests/conftest.py
index f8993a4..9780805 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -21,11 +21,9 @@
 
 
 def get_backend(backend_name):
-    from qibo.backends.pytorch import (  # the qiboml pytorch is not updated
-        PyTorchBackend,
-    )
 
     from qiboml.backends.jax import JaxBackend
+    from qiboml.backends.pytorch import PyTorchBackend
     from qiboml.backends.tensorflow import TensorflowBackend
 
     NAME2BACKEND = {