Altered Replay dataset such that it doesn't have a vector of key ids since the iterator effectively does this anyway and compressed the representation of actions.