fix bug in deprecated collapse_wavefunction (#3156)

Fixes #3153.
quantumlib · Jul 21, 2020 · 65f2b16 · 65f2b16
1 parent ea20a18
commit 65f2b16
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/cirq/sim/clifford/clifford_simulator.py b/cirq/sim/clifford/clifford_simulator.py
@@ -428,9 +428,9 @@ def _apply_Y(self, qubit: int):
         deadline='v0.10.0',
         fix='Use collapse_state_vector instead.',
         parameter_desc='collapse_wavefunction',
-        match=lambda args, kwargs: 'collapse_wave_function' in kwargs,
-        rewrite=lambda args, kwargs: (args, {(
-            'collapse_state_vector' if k == 'collapse_wave_function' else k): v
+        match=lambda args, kwargs: 'collapse_wavefunction' in kwargs,
+        rewrite=lambda args, kwargs: (args, {('collapse_state_vector' if k ==
+                                              'collapse_wavefunction' else k): v
                                              for k, v in kwargs.items()}))
     def perform_measurement(self,
                             qubits: Sequence[ops.Qid],

diff --git a/cirq/sim/clifford/clifford_simulator_test.py b/cirq/sim/clifford/clifford_simulator_test.py
@@ -499,9 +499,9 @@ def test_deprecated():
                                   'deprecated'):
         _ = clifford_state.wave_function()
 
-    with cirq.testing.assert_logs('collapse_wave_function',
+    with cirq.testing.assert_logs('collapse_wavefunction',
                                   'collapse_state_vector', 'deprecated'):
         # pylint: disable=unexpected-keyword-arg,no-value-for-parameter
         _ = clifford_state.perform_measurement([q],
                                                prng=0,
-                                               collapse_wave_function=True)
+                                               collapse_wavefunction=True)