Values for each stage in the stage_results dictionary of the metrics_cache should be lists, where each entry in the list are epoch results. If batch level results are wanted then the entry is a list. Create a test to lock in this behaviour. Context in https://git.ichec.ie/performance/toolshed/iclearn/-/merge_requests/45