Close shuffle writers during failure & remove executorId from TaskContext