code: def argtopk_preprocess(a, idx):
      """Preparatory step for argtopk
  
      Put data together with its original indices in a tuple.
      """
      return a, idx
  
imports: from __future__ import annotations
  import contextlib
  from collections.abc import Container, Iterable, Sequence
  from functools import wraps
  from numbers import Integral
  import numpy
  from tlz import concat
  from dask.core import flatten
label_desc: Use image node_id 6 for calling a global function with example usage: argtopk_preprocess(a, idx) and returns: a,  idx