Return processor_id and project_id in get_qcs_objects_for_notebooks

cirq-google/cirq_google/engine/qcs_notebook.py

@@ -32,6 +32,8 @@ class QCSObjectsForNotebook:
     device: cirq.Device
     sampler: Union[PhasedFSimEngineSimulator, ProcessorSampler]
     signed_in: bool
+    processor_id: Optional[str]
+    project_id: Optional[str]
 
     @property
     def is_simulator(self):
@@ -90,8 +92,11 @@ def get_qcs_objects_for_notebook(
             if not processors:
                 raise ValueError("No processors available.")
             processor = processors[0]
+            processor_id = processor.processor_id
             print(f"Available processors: {[p.processor_id for p in processors]}")
             print(f"Using processor: {processor.processor_id}")
+        if not project_id:
+            project_id = processor.project_id
         device = processor.get_device()
         sampler = processor.get_sampler()
         signed_in = True
@@ -105,7 +110,13 @@ def get_qcs_objects_for_notebook(
         device = Sycamore
         signed_in = False
 
-    return QCSObjectsForNotebook(device=device, sampler=sampler, signed_in=signed_in)
+    return QCSObjectsForNotebook(
+        device=device,
+        sampler=sampler,
+        signed_in=signed_in,
+        project_id=project_id,
+        processor_id=processor_id,
+    )
 
 
 # pylint: enable=missing-raises-doc

cirq-google/cirq_google/engine/qcs_notebook_test.py

@@ -17,11 +17,13 @@
 
 
 def test_get_device_sampler():
-    result = get_qcs_objects_for_notebook()
-    assert result.device is cg.Sycamore
+    result = get_qcs_objects_for_notebook('not_a_valid_project_name')
+    assert isinstance(result.device, cg.GridDevice)
     assert not result.signed_in
     assert isinstance(result.sampler, cg.PhasedFSimEngineSimulator)
     assert result.is_simulator
 
+    # Note: if running locally with application default credentials,
+    # you actually will be signed_in
     result = get_qcs_objects_for_notebook("", "")
-    assert not result.signed_in
+    assert isinstance(result.device, cg.GridDevice)