code: def _time_query(bm, ts):
      now = time.perf_counter()
      for _ in range(reps):
          for ta, tb in zip(ts[:-1], ts[1:]):
              if ta > tb:
                  ta, tb = tb, ta
              bm(ta, tb)
      return time.perf_counter() - now
  
imports: import argparse
  import logging
  import os
  import time
  import numpy.random
  import torch
  import torchsde
  from diagnostics import utils
label_desc: Use image node_id 1 for calling a global function with example usage: _time_query(bm, ts) and returns: unknown