code: def check_forward_output(self, output: Any):
      assert isinstance(output, Tensor)
      assert output.shape == torch.Size([1, 384])
  
file_path: lightning-flash/tests/text/embedding/test_model.py
imports: import os
  from typing import Any
  import flash
  import pytest
  import torch
  from flash.core.utilities.imports import _TOPIC_TEXT_AVAILABLE
  from flash.text import TextClassificationData, TextEmbedder
  from torch import Tensor
  from tests.helpers.task_tester import TaskTester
label_desc: Use image node_id 2 for calling the TestTextEmbedder obj's underlying member method code with example usage: obj.check_forward_output(output) without return types