fix: importing pytorch backend from qiboml in the tests too

qiboteam · Nov 19, 2024 · a24fd05 · a24fd05
1 parent cf6f80c
commit a24fd05
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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 = {