pyqrack-cpu 1.44.8__py3-none-win_amd64.whl → 1.44.9__py3-none-win_amd64.whl

pyqrack/qrack_circuit.py

@@ -45,13 +45,22 @@ class QrackCircuit:
         cid(int): Qrack ID of this circuit
     """
 
-    def __init__(self, is_collapse = True, is_near_clifford = False, clone_cid = -1, is_inverse=False, past_light_cone = []):
+    def __init__(
+        self,
+        is_collapse=True,
+        is_near_clifford=False,
+        clone_cid=-1,
+        is_inverse=False,
+        past_light_cone=[],
+    ):
         if clone_cid < 0:
             self.cid = Qrack.qrack_lib.init_qcircuit(is_collapse, is_near_clifford)
         elif is_inverse:
             self.cid = Qrack.qrack_lib.qcircuit_inverse(clone_cid)
         elif len(past_light_cone) > 0:
-            self.cid = Qrack.qrack_lib.qcircuit_past_light_cone(clone_cid, len(past_light_cone), self._ulonglong_byref(past_light_cone))
+            self.cid = Qrack.qrack_lib.qcircuit_past_light_cone(
+                clone_cid, len(past_light_cone), self._ulonglong_byref(past_light_cone)
+            )
         else:
             self.cid = Qrack.qrack_lib.init_qcircuit_clone(clone_cid)
 
@@ -74,7 +83,7 @@ class QrackCircuit:
         nrm = abs(m[0])
         if (nrm * nrm) < sys.float_info.epsilon:
             phase = 1.0 + 0.0j
-            th = math.pi;
+            th = math.pi
         else:
             phase = m[0] / nrm
             if nrm > 1.0:
@@ -123,7 +132,7 @@ class QrackCircuit:
         Raises:
             RuntimeError: QrackCircuit C++ library raised an exception.
         """
-        return QrackCircuit(clone_cid = self.cid, is_inverse = False)
+        return QrackCircuit(clone_cid=self.cid, is_inverse=False)
 
     def inverse(self):
         """Make a new circuit that is the exact inverse of this circuit
@@ -131,7 +140,7 @@ class QrackCircuit:
         Raises:
             RuntimeError: QrackCircuit C++ library raised an exception.
         """
-        return QrackCircuit(clone_cid = self.cid, is_inverse = True)
+        return QrackCircuit(clone_cid=self.cid, is_inverse=True)
 
     def past_light_cone(self, q):
         """Make a new circuit with just this circuits' past light cone for certain qubits.
@@ -142,7 +151,7 @@ class QrackCircuit:
         Raises:
             RuntimeError: QrackCircuit C++ library raised an exception.
         """
-        return QrackCircuit(clone_cid = self.cid, is_inverse = False, past_light_cone = q)
+        return QrackCircuit(clone_cid=self.cid, is_inverse=False, past_light_cone=q)
 
     def get_qubit_count(self):
         """Get count of qubits in circuit
@@ -178,7 +187,9 @@ class QrackCircuit:
             RuntimeError: QrackSimulator raised an exception.
         """
         if len(m) < 4:
-            raise ValueError("2x2 matrix 'm' in QrackCircuit.mtrx() must contain at least 4 elements.")
+            raise ValueError(
+                "2x2 matrix 'm' in QrackCircuit.mtrx() must contain at least 4 elements."
+            )
         Qrack.qrack_lib.qcircuit_append_1qb(self.cid, self._complex_byref(m), q)
 
     def ucmtrx(self, c, m, q, p):
@@ -199,7 +210,9 @@ class QrackCircuit:
             RuntimeError: QrackSimulator raised an exception.
         """
         if len(m) < 4:
-            raise ValueError("2x2 matrix 'm' in QrackCircuit.ucmtrx() must contain at least 4 elements.")
+            raise ValueError(
+                "2x2 matrix 'm' in QrackCircuit.ucmtrx() must contain at least 4 elements."
+            )
         Qrack.qrack_lib.qcircuit_append_mc(
             self.cid, self._complex_byref(m), len(c), self._ulonglong_byref(c), q, p
         )
@@ -233,7 +246,7 @@ class QrackCircuit:
         Args:
             filename: Name of file
         """
-        Qrack.qrack_lib.qcircuit_out_to_file(self.cid, filename.encode('utf-8'))
+        Qrack.qrack_lib.qcircuit_out_to_file(self.cid, filename.encode("utf-8"))
 
     def in_from_file(filename):
         """Read in optimized circuit from file
@@ -245,7 +258,7 @@ class QrackCircuit:
             filename: Name of file
         """
         out = QrackCircuit()
-        Qrack.qrack_lib.qcircuit_in_from_file(out.cid, filename.encode('utf-8'))
+        Qrack.qrack_lib.qcircuit_in_from_file(out.cid, filename.encode("utf-8"))
 
         return out
 
@@ -269,7 +282,7 @@ class QrackCircuit:
             filename: Name of file
         """
         tokens = []
-        with open(filename, 'r') as file:
+        with open(filename, "r") as file:
             tokens = file.read().split()
         return int(tokens[1])
 
@@ -309,7 +322,7 @@ class QrackCircuit:
             )
 
         tokens = []
-        with open(filename, 'r') as file:
+        with open(filename, "r") as file:
             return QrackCircuit.string_to_qiskit_circuit(file.read())
 
     def string_to_qiskit_circuit(circ_string):
@@ -361,8 +374,8 @@ class QrackCircuit:
 
                 mtrx = []
                 for _ in range(4):
-                    amp = tokens[i].replace("(","").replace(")","").split(',')
-                    mtrx.append(float(amp[0]) + float(amp[1])*1j)
+                    amp = tokens[i].replace("(", "").replace(")", "").split(",")
+                    mtrx.append(float(amp[0]) + float(amp[1]) * 1j)
                     i += 1
 
                 mtrx = QrackCircuit._make_mtrx_unitary(mtrx)
@@ -375,7 +388,7 @@ class QrackCircuit:
 
                 payloads[key] = op
 
-            gate_list=[]
+            gate_list = []
             for j in range(1 << control_count):
                 if j in payloads:
                     gate_list.append(payloads[j])
@@ -421,7 +434,12 @@ class QrackCircuit:
             if o.name in ["x", "y", "z", "u"]:
                 out.mtrx(op, circ.find_bit(gate.qubits[0])[0])
             else:
-                out.ucmtrx([circ.find_bit(gate.qubits[0])[0]], op, circ.find_bit(gate.qubits[1])[0], 1)
+                out.ucmtrx(
+                    [circ.find_bit(gate.qubits[0])[0]],
+                    op,
+                    circ.find_bit(gate.qubits[1])[0],
+                    1,
+                )
 
         return out
 
@@ -431,9 +449,9 @@ class QrackCircuit:
         psi0=None,
         gate_opts=None,
         tags=None,
-        psi0_dtype='complex128',
-        psi0_tag='PSI0',
-        bra_site_ind_id='b{}'
+        psi0_dtype="complex128",
+        psi0_tag="PSI0",
+        bra_site_ind_id="b{}",
     ):
         """Convert an output file to a Quimb circuit
 
@@ -464,25 +482,32 @@ class QrackCircuit:
         basis_gates = ["u", "cx"]
         qcirc = transpile(qcirc, basis_gates=basis_gates, optimization_level=3)
 
-        tcirc = qtn.Circuit(
-            N=qcirc.num_qubits,
-            psi0=psi0,
-            gate_opts=gate_opts,
-            tags=tags,
-            psi0_dtype=psi0_dtype,
-            psi0_tag=psi0_tag,
-            bra_site_ind_id=bra_site_ind_id
-        ) if circuit_type == QuimbCircuitType.Circuit else (
-            qtn.CircuitDense(N=qcirc.num_qubits, psi0=psi0, gate_opts=gate_opts, tags=tags) if circuit_type == QuimbCircuitType.CircuitDense else
-                qtn.CircuitMPS(
+        tcirc = (
+            qtn.Circuit(
+                N=qcirc.num_qubits,
+                psi0=psi0,
+                gate_opts=gate_opts,
+                tags=tags,
+                psi0_dtype=psi0_dtype,
+                psi0_tag=psi0_tag,
+                bra_site_ind_id=bra_site_ind_id,
+            )
+            if circuit_type == QuimbCircuitType.Circuit
+            else (
+                qtn.CircuitDense(
+                    N=qcirc.num_qubits, psi0=psi0, gate_opts=gate_opts, tags=tags
+                )
+                if circuit_type == QuimbCircuitType.CircuitDense
+                else qtn.CircuitMPS(
                     N=qcirc.num_qubits,
                     psi0=psi0,
                     gate_opts=gate_opts,
                     tags=tags,
                     psi0_dtype=psi0_dtype,
                     psi0_tag=psi0_tag,
-                    bra_site_ind_id=bra_site_ind_id
+                    bra_site_ind_id=bra_site_ind_id,
                 )
+            )
         )
         for gate in qcirc.data:
             o = gate.operation
@@ -491,17 +516,18 @@ class QrackCircuit:
                 ph = float(o.params[1])
                 lm = float(o.params[2])
 
-                tcirc.apply_gate('U3', th, ph, lm, qcirc.find_bit(gate.qubits[0])[0])
+                tcirc.apply_gate("U3", th, ph, lm, qcirc.find_bit(gate.qubits[0])[0])
             else:
-                tcirc.apply_gate('CNOT', qcirc.find_bit(gate.qubits[0])[0], qcirc.find_bit(gate.qubits[1])[0])
+                tcirc.apply_gate(
+                    "CNOT",
+                    qcirc.find_bit(gate.qubits[0])[0],
+                    qcirc.find_bit(gate.qubits[1])[0],
+                )
 
         return tcirc
 
     def file_to_tensorcircuit(
-        filename,
-        inputs=None,
-        circuit_params=None,
-        binding_params=None
+        filename, inputs=None, circuit_params=None, binding_params=None
     ):
         """Convert an output file to a TensorCircuit circuit
 
@@ -528,9 +554,11 @@ class QrackCircuit:
         basis_gates = ["u", "cx"]
         qcirc = transpile(qcirc, basis_gates=basis_gates, optimization_level=3)
 
-        return tc.Circuit.from_qiskit(qcirc, qcirc.num_qubits, inputs, circuit_params, binding_params)
+        return tc.Circuit.from_qiskit(
+            qcirc, qcirc.num_qubits, inputs, circuit_params, binding_params
+        )
 
-    def in_from_tensorcircuit(tcirc, enable_instruction = False, enable_inputs = False):
+    def in_from_tensorcircuit(tcirc, enable_instruction=False, enable_inputs=False):
         """Convert a TensorCircuit circuit to a QrackCircuit
 
         Accepts a TensorCircuit circuit and outputs an equivalent QrackCircuit

pyqrack/qrack_neuron.py

@@ -46,10 +46,10 @@ class QrackNeuron:
         simulator,
         controls,
         target,
-        activation_fn = NeuronActivationFn.Sigmoid,
-        alpha = 1.0,
-        tolerance = sys.float_info.epsilon,
-        _init = True
+        activation_fn=NeuronActivationFn.Sigmoid,
+        alpha=1.0,
+        tolerance=sys.float_info.epsilon,
+        _init=True,
     ):
         self.simulator = simulator
         self.controls = controls
@@ -61,7 +61,15 @@ class QrackNeuron:
         if not _init:
             return
 
-        self.nid = Qrack.qrack_lib.init_qneuron(simulator.sid, len(controls), self._ulonglong_byref(controls), target, activation_fn, alpha, tolerance)
+        self.nid = Qrack.qrack_lib.init_qneuron(
+            simulator.sid,
+            len(controls),
+            self._ulonglong_byref(controls),
+            target,
+            activation_fn,
+            alpha,
+            tolerance,
+        )
 
         self._throw_if_error()
 
@@ -79,7 +87,14 @@ class QrackNeuron:
         Raises:
             RuntimeError: QrackNeuron C++ library raised an exception.
         """
-        result = QrackNeuron(self.simulator, self.controls, self.target, self.activation_fn, self.alpha, self.tolerance)
+        result = QrackNeuron(
+            self.simulator,
+            self.controls,
+            self.target,
+            self.activation_fn,
+            self.alpha,
+            self.tolerance,
+        )
         self.nid = Qrack.qrack_lib.clone_qneuron(self.simulator.sid)
         self._throw_if_error()
         return result
@@ -107,7 +122,9 @@ class QrackNeuron:
             RuntimeError: QrackSimulator raised an exception.
         """
         if len(a) < (1 << len(self.controls)):
-            raise ValueError("Angles 'a' in QrackNeuron.set_angles() must contain at least (2 ** len(self.controls)) elements.")
+            raise ValueError(
+                "Angles 'a' in QrackNeuron.set_angles() must contain at least (2 ** len(self.controls)) elements."
+            )
         Qrack.qrack_lib.set_qneuron_angles(self.nid, self._real1_byref(a))
         self._throw_if_error()
 
@@ -145,7 +162,7 @@ class QrackNeuron:
 
         Nonlinear activation functions can be important to neural net
         applications, like DNN. The available activation functions are
-        enumerated in `NeuronActivationFn`. 
+        enumerated in `NeuronActivationFn`.
 
         Raises:
             RuntimeError: QrackNeuron C++ library raised an exception.

pyqrack/qrack_neuron_torch_layer.py

@@ -24,7 +24,7 @@ from itertools import chain, combinations
 def powerset(iterable):
     "powerset([1,2,3]) --> () (1,) (2,) (3,) (1,2) (1,3) (2,3,) (1,2,3)"
     s = list(iterable)
-    return chain.from_iterable(combinations(s, r) for r in range(len(s)+1))
+    return chain.from_iterable(combinations(s, r) for r in range(len(s) + 1))
 
 
 class QrackTorchNeuron(nn.Module if _IS_TORCH_AVAILABLE else object):
@@ -33,6 +33,7 @@ class QrackTorchNeuron(nn.Module if _IS_TORCH_AVAILABLE else object):
     Attributes:
         neuron(QrackNeuron): QrackNeuron backing this torch wrapper
     """
+
     def __init__(self, neuron: QrackNeuron):
         super().__init__()
         self.neuron = neuron
@@ -46,6 +47,7 @@ class QrackTorchNeuron(nn.Module if _IS_TORCH_AVAILABLE else object):
 
 class QrackNeuronFunction(Function if _IS_TORCH_AVAILABLE else object):
     """Static forward/backward/apply functions for QrackTorchNeuron"""
+
     @staticmethod
     def forward(ctx, neuron):
         # Save for backward
@@ -56,7 +58,11 @@ class QrackNeuronFunction(Function if _IS_TORCH_AVAILABLE else object):
         final_prob = neuron.simulator.prob(neuron.target)
         ctx.delta = final_prob - init_prob
 
-        return torch.tensor([ctx.delta], dtype=torch.float32) if _IS_TORCH_AVAILABLE else ctx.delta
+        return (
+            torch.tensor([ctx.delta], dtype=torch.float32)
+            if _IS_TORCH_AVAILABLE
+            else ctx.delta
+        )
 
     @staticmethod
     def backward(ctx, grad_output):
@@ -69,13 +75,22 @@ class QrackNeuronFunction(Function if _IS_TORCH_AVAILABLE else object):
 
         grad = reverse_delta - ctx.delta
 
-        return torch.tensor([grad], dtype=torch.float32) if _IS_TORCH_AVAILABLE else grad
+        return (
+            torch.tensor([grad], dtype=torch.float32) if _IS_TORCH_AVAILABLE else grad
+        )
 
 
 class QrackNeuronTorchLayer(nn.Module if _IS_TORCH_AVAILABLE else object):
     """Torch layer wrapper for QrackNeuron (with power set of neurons between inputs and outputs)"""
-    def __init__(self, simulator, input_indices, output_indices,
-                 activation=int(NeuronActivationFn.Generalized_Logistic), parameters=None):
+
+    def __init__(
+        self,
+        simulator,
+        input_indices,
+        output_indices,
+        activation=int(NeuronActivationFn.Generalized_Logistic),
+        parameters=None,
+    ):
         """
         Initialize a QrackNeuron layer for PyTorch with a power set of neurons connecting inputs to outputs.
         The inputs and outputs must take the form of discrete, binary features (loaded manually into the backing QrackSimulator)
@@ -92,26 +107,40 @@ class QrackNeuronTorchLayer(nn.Module if _IS_TORCH_AVAILABLE else object):
         self.input_indices = input_indices
         self.output_indices = output_indices
         self.activation = NeuronActivationFn(activation)
-        self.fn = QrackNeuronFunction.apply if _IS_TORCH_AVAILABLE else lambda x: QrackNeuronFunction.forward(object(), x)
+        self.fn = (
+            QrackNeuronFunction.apply
+            if _IS_TORCH_AVAILABLE
+            else lambda x: QrackNeuronFunction.forward(object(), x)
+        )
 
         # Create neurons from all powerset input combinations, projecting to coherent output qubits
-        self.neurons = nn.ModuleList([
-            QrackTorchNeuron(QrackNeuron(simulator, list(input_subset), output_id, activation))
-            for input_subset in powerset(input_indices)
-            for output_id in output_indices
-        ])
+        self.neurons = nn.ModuleList(
+            [
+                QrackTorchNeuron(
+                    QrackNeuron(simulator, list(input_subset), output_id, activation)
+                )
+                for input_subset in powerset(input_indices)
+                for output_id in output_indices
+            ]
+        )
 
         # Set Qrack's internal parameters:
         param_count = 0
         for neuron_wrapper in self.neurons:
             neuron = neuron_wrapper.neuron
             p_count = 1 << len(neuron.controls)
-            neuron.set_angles(parameters[param_count:(param_count+p_count+1)] if parameters else ([0.0] * p_count))
+            neuron.set_angles(
+                parameters[param_count : (param_count + p_count + 1)]
+                if parameters
+                else ([0.0] * p_count)
+            )
             param_count += p_count
 
         self.weights = nn.ParameterList()
         for pid in range(param_count):
-            self.weights.append(nn.Parameter(torch.tensor(parameters[pid] if parameters else 0.0)))
+            self.weights.append(
+                nn.Parameter(torch.tensor(parameters[pid] if parameters else 0.0))
+            )
 
     def forward(self, _):
         # Assume quantum outputs should overwrite the simulator state
@@ -125,7 +154,9 @@ class QrackNeuronTorchLayer(nn.Module if _IS_TORCH_AVAILABLE else object):
         for neuron_wrapper in self.neurons:
             neuron = neuron_wrapper.neuron
             p_count = 1 << len(neuron.controls)
-            angles = [w.item() for w in self.weights[param_count:(param_count+p_count)]]
+            angles = [
+                w.item() for w in self.weights[param_count : (param_count + p_count)]
+            ]
             neuron.set_angles(angles)
             param_count += p_count
 
@@ -134,10 +165,6 @@ class QrackNeuronTorchLayer(nn.Module if _IS_TORCH_AVAILABLE else object):
             self.fn(neuron_wrapper.neuron)
 
         # These are classical views over quantum state; simulator still maintains full coherence
-        outputs = [
-            self.simulator.prob(output_id)
-            for output_id in self.output_indices
-        ]
+        outputs = [self.simulator.prob(output_id) for output_id in self.output_indices]
 
         return torch.tensor(outputs, dtype=torch.float32)
-