Instead of saving the simulation data imported from the workflow.hdf5 file as a python dictionary .pkl file, it would be better to only import the data required for each calculation plotting cell in the notebooks with h5py's h5.get function. See stackoverflow about this here.