code: def predict_task_proba(self, X, t=0, **kwargs):
          
          return self.predict_proba(X, **kwargs)[t]
docs: Predicts probabilistic labels for an input X on task t
  
          Args:
              X: The input for the predict_proba method
              t: The task index to predict for which to predict probabilities
          Returns:
              An [n, K_t] tensor of predictions for task t
          NOTE: By default, this method calls predict_proba and extracts element
          t. If it is possible to predict individual tasks in isolation, however,
          this method may be overriden for efficiency's sake.