Update to avoid warning

JuliaQUBO · May 28, 2024 · b6b75f0 · b6b75f0
1 parent 2c4ac44
commit b6b75f0
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/src/QAOA.jl b/src/QAOA.jl
@@ -143,8 +143,12 @@ function retrieve(
         end
     end
 
-    session = qiskit_ibm_runtime.Session(service=service, backend=backend)
-    estimator = qiskit_ibm_runtime.EstimatorV2(session=session)
+    estimator = if is_local || ibm_backend == "ibmq_qasm_simulator"
+        qiskit_ibm_runtime.EstimatorV2(backend = backend)
+    else
+        session = qiskit_ibm_runtime.Session(service=service, backend=backend)
+        qiskit_ibm_runtime.EstimatorV2(session=session)
+    end
     if !is_local
         estimator.options.default_shots = num_reads
     end

diff --git a/src/VQE.jl b/src/VQE.jl
@@ -137,8 +137,12 @@ function retrieve(
         end
     end
 
-    session = qiskit_ibm_runtime.Session(service=service, backend=backend)
-    estimator = qiskit_ibm_runtime.EstimatorV2(session=session)
+    estimator = if is_local || ibm_backend == "ibmq_qasm_simulator"
+        qiskit_ibm_runtime.EstimatorV2(backend = backend)
+    else
+        session = qiskit_ibm_runtime.Session(service=service, backend=backend)
+        qiskit_ibm_runtime.EstimatorV2(session=session)
+    end
     if !is_local
         estimator.options.default_shots = num_reads
     end