Use SampleMPs in dynamic_one_shot

[vincentmr]

Before submitting

Please complete the following checklist when submitting a PR:

• All new features must include a unit test.
  If you've fixed a bug or added code that should be tested, add a test to the
  test directory!

• All new functions and code must be clearly commented and documented.
  If you do make documentation changes, make sure that the docs build and
  render correctly by running make docs.

• Ensure that the test suite passes, by running make test.

• Add a new entry to the doc/releases/changelog-dev.md file, summarizing the
  change, and including a link back to the PR.

• The PennyLane source code conforms to
  PEP8 standards.
  We check all of our code against Pylint.
  To lint modified files, simply pip install pylint, and then
  run pylint pennylane/path/to/file.py.

When all the above are checked, delete everything above the dashed
line and fill in the pull request template.

---

Context:
The native MCM workflow breaks the device API where a sequence of MeasurementProcess objects is expected in the output.

Description of the Change:
Introduce SampleMPs in the auxiliary tape. Pass the mid_measurements dictionary around simulate and sampling to return the correct sample measurements. Modify the dynamic_one_shot transform post-processing function accordingly.

Benefits:
Conform to current API.
Road to jax.jit support.

Possible Drawbacks:
Ad hoc post-processing required in measure_with_samples.

Related GitHub Issues:
[sc-60945]

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 99.68%. Comparing base (d03216e) to head (7125646).
Report is 5 commits behind head on master.

Additional details and impacted files

@@            Coverage Diff             @@
##           master    #5486      +/-   ##
==========================================
+ Coverage   99.67%   99.68%   +0.01%     
==========================================
  Files         407      409       +2     
  Lines       38019    37907     -112     
==========================================
- Hits        37895    37789     -106     
+ Misses        124      118       -6     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[albi3ro]

All the source code looks good to me.

I'm surprised no tests failed with this change. Could we perhaps add some more unit tests to tests/transforms/test_dynamic_one_shot.py to assert the output batch when applied to a tape looks correct?

[mudit2812]

Looks good to me. Approving since the only thing I can see that needs to be changed is the additional tests Christina suggested.

[vincentmr]

Looks good to me. Approving since the only thing I can see that needs to be changed is the additional tests Christina suggested.

Thanks @mudit2812 . I added some tests in tests/transforms/test_dynamic_one_shot.py as suggested by @albi3ro . Do you think there is enough there?

[mudit2812]

The added tests seem sound to me. Could you also add a test to check the correctness for broadcasted tapes? The number of tapes in that case should be batch_size * n_shots

[albi3ro]

Thanks 🚀 !