code: def load_sample(self, sample: Dict[str, Any]) -> Dict[str, Any]:
      if DataKeys.TARGET in sample:
          sample[DataKeys.TARGET] = sample[DataKeys.TARGET].numpy()
      return super().load_sample(sample)
  
imports: import os
  from typing import Any, Dict, List, Optional, Tuple, Union
  import numpy
  from flash.core.data.io.input import DataKeys, Input
  from flash.core.data.utilities.loading import IMG_EXTENSIONS, NP_EXTENSIONS, load_image
  from flash.core.data.utilities.paths import PATH_TYPE, filter_valid_files
  from flash.core.data.utilities.samples import to_samples
  from flash.core.integrations.fiftyone.utils import FiftyOneLabelUtilities
  from flash.core.utilities.imports import _FIFTYONE_AVAILABLE, lazy_import
  from flash.image.data import ImageFilesInput, ImageNumpyInput, ImageTensorInput
  from flash.image.segmentation.output import SegmentationLabelsOutput
label_desc: Use image node_id 2 for calling the SemanticSegmentationTensorInput obj's underlying member method code with example usage: obj.load_sample(sample) and returns: super