Fix for Jupyter plotting

[ericspod]

Signed-off-by: Eric Kerfoot eric.kerfoot@kcl.ac.uk

Fixes # .

Description

This fixes an issue with the plotting in the Jupyter utilities relating to collated or decollated batches and having batches greater than 1.

Status

Work in progress

Types of changes

• Non-breaking change (fix or new feature that would not break existing functionality).
• Breaking change (fix or new feature that would cause existing functionality to change).
• New tests added to cover the changes.
• Integration tests passed locally by running ./runtests.sh -f -u --net --coverage.
• Quick tests passed locally by running ./runtests.sh --quick --unittests.
• In-line docstrings updated.
• Documentation updated, tested make html command in the docs/ folder.

[ericspod]

The current test fail looks like it has something to do with Matplotlib changing how it renders things since the content of the stored and generated images is the same, there are not changes to the plotting functions in this PR.

[wyli]

change the batch size will also change the loss curves, the expected image needs an update to match the new curve.

[ericspod]

There was a bit of variation in how the text was rendered as well and spacing somehow also changed. After accounting for the batch size these differences are still there.