code: def example_train_sample(self):
      return {
          DataKeys.INPUT: torch.rand(3, 224, 224),
          DataKeys.TARGET: {
              "bboxes": [
                  {"xmin": 10, "ymin": 10, "width": 20, "height": 20},
                  {"xmin": 30, "ymin": 30, "width": 40, "height": 40},
              ],
              "labels": [0, 1],
              "keypoints": [
                  [{"x": 10, "y": 10, "visible": 1}],
                  [{"x": 10, "y": 10, "visible": 1}],
              ],
          },
      }
  
imports: import collections
  import json
  import os
  from typing import Any
  import numpy
  import pytest
  import torch
  from flash import Trainer
  from flash.core.data.io.input import DataKeys
  from flash.core.utilities.imports import _ICEDATA_AVAILABLE, _ICEVISION_AVAILABLE, _TOPIC_IMAGE_AVAILABLE
  from flash.image import KeypointDetectionData, KeypointDetector
  from tests.helpers.task_tester import TaskTester
label_desc: Use image node_id 3 for calling the TestKeypointDetector obj's underlying member method code with example usage: obj.example_train_sample() and returns: dict