Skip to content

Commit

Permalink
Fix typing for Google measurements functions (#4450)
Browse files Browse the repository at this point in the history
- Use more general AbstractCircuit instead of Circuit.
  • Loading branch information
dstrain115 committed Aug 21, 2021
1 parent 82a7698 commit 1532975
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions cirq-google/cirq_google/api/v2/results.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ class MeasureInfo:

# TODO(#3388) Add documentation for Raises.
# pylint: disable=missing-raises-doc
def find_measurements(program: cirq.Circuit) -> List[MeasureInfo]:
def find_measurements(program: cirq.AbstractCircuit) -> List[MeasureInfo]:
"""Find measurements in the given program (circuit).
Returns:
Expand All @@ -64,7 +64,7 @@ def find_measurements(program: cirq.Circuit) -> List[MeasureInfo]:
measurements: List[MeasureInfo] = []
keys: Set[str] = set()

if isinstance(program, cirq.Circuit):
if isinstance(program, cirq.AbstractCircuit):
measure_iter = _circuit_measurements(program)
else:
raise NotImplementedError(f'Unrecognized program type: {type(program)}')
Expand All @@ -79,7 +79,7 @@ def find_measurements(program: cirq.Circuit) -> List[MeasureInfo]:


# pylint: enable=missing-raises-doc
def _circuit_measurements(circuit: cirq.Circuit) -> Iterator[MeasureInfo]:
def _circuit_measurements(circuit: cirq.AbstractCircuit) -> Iterator[MeasureInfo]:
for i, moment in enumerate(circuit):
for op in moment:
if isinstance(op.gate, cirq.MeasurementGate):
Expand Down

0 comments on commit 1532975

Please sign in to comment.