remove default values of InternalGate (#6330)

quantumlib · Oct 26, 2023 · 1549785 · 1549785
1 parent 235a0c6
commit 1549785
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 13 deletions.
diff --git a/cirq-google/cirq_google/json_test_data/InternalGate.json b/cirq-google/cirq_google/json_test_data/InternalGate.json
@@ -1,7 +1,7 @@
 {
     "cirq_type": "InternalGate",
     "gate_name": "CouplerDelayZ",
-    "gate_module": "pyle.cirqtools.pyle_gates",
+    "gate_module": "internal.module",
     "num_qubits": 2,
     "delay": 1,
     "zpa": 0.0,

diff --git a/cirq-google/cirq_google/json_test_data/InternalGate.repr b/cirq-google/cirq_google/json_test_data/InternalGate.repr
@@ -1 +1 @@
-cirq_google.InternalGate(gate_name="CouplerDelayZ", gate_module="pyle.cirqtools.pyle_gates", num_qubits=2, delay=1, zpa=0.0, zpl=None)
+cirq_google.InternalGate(gate_name="CouplerDelayZ", gate_module="internal.module", num_qubits=2, delay=1, zpa=0.0, zpl=None)
diff --git a/cirq-google/cirq_google/ops/internal_gate.py b/cirq-google/cirq_google/ops/internal_gate.py
@@ -25,18 +25,12 @@ class InternalGate(ops.Gate):
     constructor stored in `self.gate_args`.
     """
 
-    def __init__(
-        self,
-        gate_name: str,
-        gate_module: str = 'pyle.cirqtools.pyle_gates',
-        num_qubits: int = 1,
-        **kwargs,
-    ):
+    def __init__(self, gate_name: str, gate_module: str, num_qubits: int = 1, **kwargs):
         """Instatiates an InternalGate.
 
         Arguments:
             gate_name: Gate class name.
-            gate_module: The module of the gate (defualt: pyle.cirqtools.pyle_gates).
+            gate_module: The module of the gate.
             num_qubits: Number of qubits that the gate acts on.
             **kwargs: The named arguments to be passed to the gate constructor.
         """

diff --git a/cirq-google/cirq_google/ops/internal_gate_test.py b/cirq-google/cirq_google/ops/internal_gate_test.py
@@ -20,16 +20,16 @@
 def test_internal_gate():
     g = cirq_google.InternalGate(
         gate_name="CouplerDelayZ",
-        gate_module='pyle.cirqtools.pyle_gates',
+        gate_module='internal_module',
         num_qubits=2,
         delay=1,
         zpa=0.0,
         zpl=None,
     )
-    assert str(g) == 'pyle.cirqtools.pyle_gates.CouplerDelayZ(delay=1, zpa=0.0, zpl=None)'
+    assert str(g) == 'internal_module.CouplerDelayZ(delay=1, zpa=0.0, zpl=None)'
     want_repr = (
         "cirq_google.InternalGate(gate_name='CouplerDelayZ', "
-        "gate_module='pyle.cirqtools.pyle_gates', num_qubits=2, "
+        "gate_module='internal_module', num_qubits=2, "
         "delay=1, zpa=0.0, zpl=None)"
     )
     assert repr(g) == want_repr