From 3936c438e46069c5f3c105b56fb4509cea2da74c Mon Sep 17 00:00:00 2001 From: Ikko Hamamura Date: Fri, 7 Oct 2022 13:15:19 +0900 Subject: [PATCH] Add shots to QuasiDistributions in primitives.Sampler (#8791) * add shots to QuasiDistributions * add tests Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com> --- qiskit/primitives/sampler.py | 2 +- test/python/primitives/test_sampler.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/qiskit/primitives/sampler.py b/qiskit/primitives/sampler.py index 4452684d1ceb..fc5cab5607d6 100644 --- a/qiskit/primitives/sampler.py +++ b/qiskit/primitives/sampler.py @@ -128,7 +128,7 @@ def _call( ] for metadatum in metadata: metadatum["shots"] = shots - quasis = [QuasiDistribution(dict(enumerate(p))) for p in probabilities] + quasis = [QuasiDistribution(dict(enumerate(p)), shots=shots) for p in probabilities] return SamplerResult(quasis, metadata) diff --git a/test/python/primitives/test_sampler.py b/test/python/primitives/test_sampler.py index 1b35c09cf9ba..c7a3b4915bcf 100644 --- a/test/python/primitives/test_sampler.py +++ b/test/python/primitives/test_sampler.py @@ -447,6 +447,7 @@ def test_with_shots_option(self): sampler = Sampler(circuits=self._pqc) result = sampler(circuits=[0], parameter_values=params, shots=1024, seed=15) self._compare_probs(result.quasi_dists, target) + self.assertEqual(result.quasi_dists[0].shots, 1024) def test_with_shots_option_none(self): """test with shots=None option. Seed is ignored then.""" @@ -739,6 +740,7 @@ def test_options(self): params, target = self._generate_params_target([1]) result = sampler.run([self._pqc], parameter_values=params).result() self._compare_probs(result.quasi_dists, target) + self.assertEqual(result.quasi_dists[0].shots, 1024) def test_different_circuits(self): """Test collision of quantum circuits."""