code: def check_forward_output(self, output: Any):
      assert isinstance(output["prediction"], Tensor)
      assert output["prediction"].shape == torch.Size([2, 20])
  
file_path: lightning-flash/tests/tabular/forecasting/test_model.py
imports: from typing import Any
  import flash
  import pytest
  import torch
  from flash import DataKeys
  from flash.core.utilities.imports import _TOPIC_TABULAR_AVAILABLE
  from flash.tabular.forecasting import TabularForecaster
  from torch import Tensor
  from tests.helpers.task_tester import StaticDataset, TaskTester
label_desc: Use image node_id 2 for calling the TestTabularForecaster obj's underlying member method code with example usage: obj.check_forward_output(output) without return types