TaskAttemptContext has become an interface