Merge pull request #52 from qiboteam/torch_migr_fix

Importing pytorch backend from qiboml
qiboteam · Nov 19, 2024 · 3e52004 · 3e52004
2 parents 800e800 + a24fd05
commit 3e52004
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/src/qiboml/backends/__init__.py b/src/qiboml/backends/__init__.py
@@ -1,9 +1,9 @@
 from typing import Union
 
-from qibo.backends.pytorch import PyTorchBackend  # qiboml pytorch is not updated
 from qibo.config import raise_error
 
 from qiboml.backends.jax import JaxBackend
+from qiboml.backends.pytorch import PyTorchBackend
 from qiboml.backends.tensorflow import TensorflowBackend
 
 PLATFORMS = ["tensorflow", "pytorch", "jax"]

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 = {