Deprecate json_serializable_dataclass (quantumlib#5208)

Fixes quantumlib#4460
PhaseCraft · Apr 7, 2022 · 51c399c · 51c399c
1 parent c5db518
commit 51c399c
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 21 deletions.
diff --git a/cirq/experiments/cross_entropy_benchmarking.py b/cirq/experiments/cross_entropy_benchmarking.py
@@ -98,7 +98,7 @@ def purity(self) -> float:
         return self.cycle_depolarization**2
 
 
-@protocols.json_serializable_dataclass(frozen=True)
+@dataclasses.dataclass(frozen=True)
 class CrossEntropyResult:
     """Results from a cross-entropy benchmarking (XEB) experiment.
 
@@ -208,6 +208,9 @@ def _from_json_dict_(cls, data, repetitions, **kwargs):
             purity_data=purity_data,
         )
 
+    def _json_dict_(self):
+        return protocols.dataclass_json_dict(self)
+
     def __repr__(self) -> str:
         args = f'data={[tuple(p) for p in self.data]!r}, repetitions={self.repetitions!r}'
         if self.purity_data is not None:

diff --git a/cirq/experiments/grid_parallel_two_qubit_xeb.py b/cirq/experiments/grid_parallel_two_qubit_xeb.py
@@ -20,7 +20,7 @@
 by executing circuits that act on many pairs simultaneously.
 """
 
-from typing import Any, Iterable, List, Optional, Sequence, TYPE_CHECKING, Tuple, cast
+from typing import Any, Dict, Iterable, List, Optional, Sequence, TYPE_CHECKING, Tuple, cast
 import collections
 from concurrent.futures import ThreadPoolExecutor
 import dataclasses
@@ -46,7 +46,6 @@
 )
 
 if TYPE_CHECKING:
-    from typing import Dict
     import cirq
 
 DEFAULT_BASE_DIR = os.path.expanduser(
@@ -101,22 +100,25 @@ def load(params: Any, base_dir: str) -> Any:
     return protocols.read_json(filename)
 
 
-@protocols.json_serializable_dataclass
+@dataclasses.dataclass
 class GridParallelXEBMetadata:
     """Metadata for a grid parallel XEB experiment.
 
     Attributes:
         data_collection_id: The data collection ID of the experiment.
     """
 
-    qubits: List['cirq.Qid']
+    qubits: Sequence['cirq.Qid']
     two_qubit_gate: 'cirq.Gate'
     num_circuits: int
     repetitions: int
-    cycles: List[int]
-    layers: List[GridInteractionLayer]
+    cycles: Sequence[int]
+    layers: Sequence[GridInteractionLayer]
     seed: Optional[int]
 
+    def _json_dict_(self):
+        return protocols.dataclass_json_dict(self)
+
     def __repr__(self) -> str:
         return (
             'cirq.experiments.grid_parallel_two_qubit_xeb.'

diff --git a/cirq/protocols/json_serialization.py b/cirq/protocols/json_serialization.py
@@ -39,7 +39,7 @@
 import sympy
 from typing_extensions import Protocol
 
-from cirq._compat import deprecated_parameter
+from cirq._compat import deprecated, deprecated_parameter
 from cirq._doc import doc_private
 from cirq.type_workarounds import NotImplementedType
 
@@ -190,6 +190,7 @@ class name via a dot (.)
 
 # Copying the Python API, whose usage of `repr` annoys pylint.
 # pylint: disable=redefined-builtin
+@deprecated(deadline='v0.15', fix='Implement _json_dict_ using cirq.dataclass_json_dict()')
 def json_serializable_dataclass(
     _cls: Optional[Type] = None,
     *,

diff --git a/cirq/protocols/json_serialization_test.py b/cirq/protocols/json_serialization_test.py
@@ -855,10 +855,14 @@ def test_pathlib_paths(tmpdir):
 
 
 def test_json_serializable_dataclass():
-    @cirq.json_serializable_dataclass
-    class MyDC:
-        q: cirq.LineQubit
-        desc: str
+    with cirq.testing.assert_deprecated(
+        "Implement _json_dict_ using cirq.dataclass_json_dict()", deadline="v0.15"
+    ):
+
+        @cirq.json_serializable_dataclass
+        class MyDC:
+            q: cirq.LineQubit
+            desc: str
 
     my_dc = MyDC(cirq.LineQubit(4), 'hi mom')
 
@@ -885,10 +889,14 @@ def custom_resolver(name):
 
 
 def test_json_serializable_dataclass_parenthesis():
-    @cirq.json_serializable_dataclass()
-    class MyDC:
-        q: cirq.LineQubit
-        desc: str
+    with cirq.testing.assert_deprecated(
+        "Implement _json_dict_ using cirq.dataclass_json_dict()", deadline="v0.15"
+    ):
+
+        @cirq.json_serializable_dataclass()
+        class MyDC:
+            q: cirq.LineQubit
+            desc: str
 
     def custom_resolver(name):
         if name == 'MyDC':
@@ -918,11 +926,15 @@ def custom_resolver(name):
 
 
 def test_json_serializable_dataclass_namespace():
-    @cirq.json_serializable_dataclass(namespace='cirq.experiments')
-    class QuantumVolumeParams:
-        width: int
-        depth: int
-        circuit_i: int
+    with cirq.testing.assert_deprecated(
+        "Implement _json_dict_ using cirq.dataclass_json_dict()", deadline="v0.15"
+    ):
+
+        @cirq.json_serializable_dataclass(namespace='cirq.experiments')
+        class QuantumVolumeParams:
+            width: int
+            depth: int
+            circuit_i: int
 
     qvp = QuantumVolumeParams(width=5, depth=5, circuit_i=0)