code: def check_forward_output(self, output: Any):
      assert isinstance(output, Tensor)
      assert output.shape == torch.Size([1, 128])
  
file_path: lightning-flash/tests/text/seq2seq/summarization/test_model.py
imports: from typing import Any
  from unittest.mock import patch
  import pytest
  import torch
  from flash import DataKeys
  from flash.core.utilities.imports import _TOPIC_SERVE_AVAILABLE, _TOPIC_TEXT_AVAILABLE
  from flash.text import SummarizationTask
  from torch import Tensor
  from tests.helpers.task_tester import TaskTester
label_desc: Use image node_id 2 for calling the TestSummarizationTask obj's underlying member method code with example usage: obj.check_forward_output(output) without return types