fix test

tencent-quantum-lab · Apr 9, 2024 · 3a74830 · 3a74830
1 parent 5c0c768
commit 3a74830
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorcircuit/backends/jax_backend.py b/tensorcircuit/backends/jax_backend.py
@@ -631,7 +631,7 @@ def is_sparse(self, a: Tensor) -> bool:
         return isinstance(a, sparse.BCOO)  # type: ignore
 
     def device(self, a: Tensor) -> str:
-        dev = a.devices()
+        (dev,) = a.devices()
         return self._dev2str(dev)
 
     def device_move(self, a: Tensor, dev: Any) -> Tensor: