revert integration/test_sampler_v2.py

Qiskit · Jun 3, 2024 · 44aeec9 · 44aeec9
1 parent 60ccaac
commit 44aeec9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/integration/test_sampler_v2.py b/test/integration/test_sampler_v2.py
@@ -23,7 +23,7 @@
 from qiskit import ClassicalRegister, QuantumCircuit, QuantumRegister, transpile
 from qiskit.circuit import Parameter
 from qiskit.circuit.library import RealAmplitudes, UnitaryGate
-from qiskit.primitives import PrimitiveResult, SamplerPubResult
+from qiskit.primitives import PrimitiveResult
 from qiskit.primitives.containers import BitArray
 from qiskit.primitives.containers.data_bin import DataBin
 from qiskit.primitives.containers.sampler_pub import SamplerPub
@@ -556,7 +556,9 @@ def _verify_result_type(self, result, num_pubs, targets=None):
         self.assertIsInstance(result.metadata, dict)
         self.assertEqual(len(result), num_pubs)
         for idx, pub_result in enumerate(result):
-            self.assertIsInstance(pub_result, SamplerPubResult)
+            # TODO: We need to update the following test to check `SamplerPubResult`
+            # when the server side is upgraded to Qiskit 1.1.
+            self.assertIsInstance(pub_result, PubResult)
             self.assertIsInstance(pub_result.data, DataBin)
             self.assertIsInstance(pub_result.metadata, dict)
             if targets: