Skip to content

Commit c80fe6f

Browse files
committed
Merge branch 'master' of github.com:tensorflow/quantum into tfq_serialize_proj
2 parents 3608a49 + 2773a6d commit c80fe6f

File tree

3 files changed

+152
-101
lines changed

3 files changed

+152
-101
lines changed

tensorflow_quantum/core/serialize/serializer.py

+7-6
Original file line numberDiff line numberDiff line change
@@ -804,8 +804,8 @@ def serialize_circuit(circuit_inp):
804804
Currently we only support scalar multiplication of symbols and
805805
no other more complex arithmetic expressions. This means
806806
we can support things like X**(3*alpha), and Rx(alpha). Because
807-
we use the `cirq.Program` proto, we only support `cirq.GridQubit` instances
808-
during serialization of circuits.
807+
we use the `cirq.Program` proto, we only support `cirq.GridQubit`
808+
and `cirq.LineQubit` instances during serialization of circuits.
809809
810810
Note: once serialized terminal measurements are removed.
811811
@@ -832,10 +832,10 @@ def serialize_circuit(circuit_inp):
832832
measured_qubits = set()
833833
for op in moment:
834834
for qubit in op.qubits:
835-
if not isinstance(qubit, cirq.GridQubit):
835+
if not isinstance(qubit, (cirq.GridQubit, cirq.LineQubit)):
836836
raise ValueError(
837837
"Attempted to serialize circuit that don't use "
838-
"only cirq.GridQubits.")
838+
"only cirq.GridQubits or cirq.LineQubits.")
839839

840840
if isinstance(op.gate, cirq.MeasurementGate):
841841
for qubit in op.qubits:
@@ -916,9 +916,10 @@ def serialize_paulisum(paulisum):
916916
raise TypeError("serialize requires a cirq.PauliSum object."
917917
" Given: " + str(type(paulisum)))
918918

919-
if any(not isinstance(qubit, cirq.GridQubit) for qubit in paulisum.qubits):
919+
if any(not isinstance(qubit, (cirq.LineQubit, cirq.GridQubit))
920+
for qubit in paulisum.qubits):
920921
raise ValueError("Attempted to serialize a paulisum that doesn't use "
921-
"only cirq.GridQubits.")
922+
"only cirq.GridQubits or cirq.LineQubits.")
922923

923924
paulisum_proto = pauli_sum_pb2.PauliSum()
924925
for term in paulisum:

0 commit comments

Comments
 (0)