A couple of bug fixes.

CQCL · Mar 11, 2024 · f6fce3b · f6fce3b
1 parent 5a0f418
commit f6fce3b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pytket/extensions/cutensornet/general_state/tensor_network_state.py b/pytket/extensions/cutensornet/general_state/tensor_network_state.py
@@ -47,7 +47,8 @@ def __init__(
             loglevel: Internal logger output level.
         """
         self._logger = set_logger("GeneralState", loglevel)
-        self._circuit = circuit.replace_implicit_wire_swaps()
+        self._circuit = circuit
+        self._circuit.replace_implicit_wire_swaps()
         self._handle = libhandle.handle
         self._dev = libhandle.dev
 

diff --git a/tests/test_general_state_operator_expval.py b/tests/test_general_state_operator_expval.py
@@ -156,7 +156,7 @@ def to_bool_tuple(n_qubits: int, x: int) -> tuple:
 
     with CuTensorNetHandle() as libhandle:
         state = GeneralState(ket_circ, libhandle)
-        oper = GeneralOperator(op, 2, libhandle)
+        oper = GeneralOperator(op, n_qubits, libhandle)
         ev = GeneralExpectationValue(state, oper, libhandle)
         ovl, state_norm = ev.configure().prepare().compute()
         ev.destroy()