Update SimulatedProcessorWithLocalDeviceRecord._get_input_device to G…

…ridDevice
quantumlib · Jun 19, 2022 · fd216b6 · fd216b6
1 parent a0e4144
commit fd216b6
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/cirq-google/cirq_google/workflow/processor_record.py b/cirq-google/cirq_google/workflow/processor_record.py
@@ -168,13 +168,12 @@ class SimulatedProcessorWithLocalDeviceRecord(SimulatedProcessorRecord):
     """
 
     def _get_input_device(self) -> 'cirq.Device':
-        """Return a `cg.SerializableDevice` for the specified processor_id.
+        """Return a `cg.GridDevice` for the specified processor_id.
 
         Only 'rainbow' and 'weber' are recognized processor_ids and the device information
         may not be up-to-date, as it is completely local.
         """
 
-        gate_sets = [cg.FSIM_GATESET]
         device_spec = _create_device_spec_from_template(MOST_RECENT_TEMPLATES[self.processor_id])
-        device = cg.SerializableDevice.from_proto(device_spec, gate_sets)
+        device = cg.GridDevice.from_proto(device_spec)
         return device