Feature: LVAE predict on whole dataset

[melisande-c]

Description

Leaving as draft for now because haven't written tests (or tested functionality elsewhere). Haven't got to grips with how to initialise the dataset correctly.

• What: Function to use LVAE to predict on whole MultiChDLoader dataset.
• Why: Need prediction functionality, leaving it as PyTorch for now because there are some complications with using the Lightning predict_step.
• How: Iterate through the dataset, use previously defined functions to predict each batch and create tile information, stitch predictions with CAREamics stitch_predictions function.

Changes Made

TBC

• Added: List new features or files added.
• Modified: Describe existing features or files modified.
• Removed: Detail features or files that were removed.

Additional Notes and Examples

Include any additional notes or context that reviewers should be aware of, including snippets of code illustrating your new feature.

---

Please ensure your PR meets the following requirements:

• Code builds and passes tests locally, including doctests
• New tests have been added (for bug fixes/features)
• Pre-commit passes
• PR to the documentation exists (for bug fixes / features)