I think it will be easier to do:
`return {'time_fig': time_fig, 'mem_fig': mem_fig, 'comp_data': data}`