Fixes for the PartitionExecutor. We will just pass around a DonePartitionsNotification internally instead of passing around its members