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

@@ -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)
 

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."""