docs: def predict_task(self, X, t=0, break_ties="random", **kwargs):
          """Predicts int labels for an input X on task t
  
          Args:
              X: The input for the predict_task_proba method
              t: The task index to predict
          Returns:
              An n-dim tensor of int predictions for the specified task
          """
          Y_tp = self.predict_task_proba(X, t=t, **kwargs)
          Y_tph = self._break_ties(Y_tp, break_ties)
          return Y_tph