code: def store_extended_prediction(
      self, data, sample, output_dir, extension
  ):
      output_dir = os.path.join(
          output_dir, filename(sample["image_path"])
      )
      if not os.path.exists(output_dir):
          os.mkdir(output_dir)
  
      super().store_extended_prediction(
          data, sample, output_dir, extension
      )
  
imports: import os
  from copy import deepcopy
  from dataclasses import dataclass, field
  from random import shuffle
  import math
  import numpy
  from paiargparse import pai_dataclass, pai_meta
  from tfaip.data.pipeline.definitions import PipelineMode, INPUT_PROCESSOR, TARGETS_PROCESSOR
  from tqdm import tqdm
  from lxml import etree
  import cv2
  from typing import List, Generator, Optional, Iterable, Dict, Any, Tuple
  from enum import IntEnum
  from calamari_ocr.ocr.dataset.datareader.base import CalamariDataGenerator, CalamariDataGeneratorParams, InputSample, SampleMeta
  from calamari_ocr.utils import split_all_ext, filename, glob_all
  from calamari_ocr.ocr.predict.params import Prediction
  import logging
label_desc: Use image node_id 5 for calling the PageXMLReader obj's underlying member method code with example usage: obj.store_extended_prediction(data, sample, output_dir, extension) without return types