Add to_dict() to KeyValueExecutableSpec (#5072)

Goal would be to fix #4734
quantumlib · Mar 17, 2022 · b2d8dd9 · b2d8dd9
1 parent c153664
commit b2d8dd9
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/cirq-google/cirq_google/workflow/quantum_executable.py b/cirq-google/cirq_google/workflow/quantum_executable.py
@@ -53,6 +53,9 @@ class KeyValueExecutableSpec(ExecutableSpec):
     executable_family: str
     key_value_pairs: Tuple[Tuple[str, Any], ...] = ()
 
+    def to_dict(self) -> Dict[str, Any]:
+        return dict(self.key_value_pairs)
+
     @classmethod
     def _json_namespace_(cls) -> str:
         return 'cirq.google'

diff --git a/cirq-google/cirq_google/workflow/quantum_executable_test.py b/cirq-google/cirq_google/workflow/quantum_executable_test.py
@@ -63,6 +63,18 @@ def test_kv_executable_spec():
         hash(KeyValueExecutableSpec(executable_family='', key_value_pairs=[('name', 'test')]))
 
 
+def test_dict_round_trip():
+    input_dict = dict(name='test', idx=5)
+
+    kv = KeyValueExecutableSpec.from_dict(
+        input_dict, executable_family='cirq_google.algo_benchmarks.example'
+    )
+
+    actual_dict = kv.to_dict()
+
+    assert input_dict == actual_dict
+
+
 def test_kv_repr():
     kv = _get_example_spec()
     cirq.testing.assert_equivalent_repr(kv, global_vals={'cirq_google': cirq_google})