code: def __init__(self, src_dir):
      self.src_dir = src_dir
  
imports: import os
  import sys
  import textwrap
  from copy import copy
  from pathlib import Path
  import pytorch_sphinx_theme
  import torchvision
  import torchvision.models
  from sphinx_gallery.sorting import ExplicitOrder
  from tabulate import tabulate
  from sphinx_gallery import gen_rst
  from docutils import nodes
  from sphinx import addnodes
  from sphinx.util.docfields import TypedField
label_desc: Use image node_id 1 to create a new CustomGalleryExampleSortKey object  with example: obj = CustomGalleryExampleSortKey(src_dir)