diff --git a/tests/conftest.py b/tests/conftest.py index 75acd9c..c5e9ed4 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -15,13 +15,10 @@ BACKENDS = ["cutensornet"] def get_backend(backend_name): - from qibotn.backends.qmatchatea import QMatchaTeaBackend from qibotn.backends.cutensornet import CuTensorNet + from qibotn.backends.qmatchatea import QMatchaTeaBackend - NAME2BACKEND = { - "qmatchatea": QMatchaTeaBackend, - "cutensornet": CuTensorNet - } + NAME2BACKEND = {"qmatchatea": QMatchaTeaBackend, "cutensornet": CuTensorNet} return NAME2BACKEND[backend_name]() diff --git a/tests/test_cuquantum_cutensor_backend.py b/tests/test_cuquantum_cutensor_backend.py index d5b2ffb..01d8d6e 100644 --- a/tests/test_cuquantum_cutensor_backend.py +++ b/tests/test_cuquantum_cutensor_backend.py @@ -1,4 +1,5 @@ import math + import cupy as cp import pytest import qibo