code: def main(args: Namespace, config: AttrDict):
      if config.SLURM.USE_SLURM:
          assert (
              is_submitit_available()
          ), "Please 'pip install submitit' to schedule jobs on SLURM"
          extract_features_and_run_knn_on_slurm(config)
      else:
          extract_features_and_run_knn(args.node_id, config)
  
file_path: vissl/tools/nearest_neighbor_test.py
imports: import sys
  from argparse import Namespace
  from typing import Any, List
  from vissl.config import AttrDict
  from vissl.utils.hydra_config import compose_hydra_configuration, convert_to_attrdict
  from vissl.utils.knn_utils import extract_features_and_run_knn, extract_features_and_run_knn_on_slurm
  from vissl.utils.slurm import is_submitit_available
label_desc: Use image node_id 1 for calling a global function with example usage: main(args, config) without return types