code: def _compare(w0, ts, msg=""):
      bm = torchsde.BrownianPath(t0=t0, w0=w0)
      bp_py_time = _time_query(bm, ts)
      logging.warning(
          f"{msg} (torchsde.BrownianPath): {bp_py_time:.4f}"
      )
  
      bm = torchsde.BrownianTree(t0=t0, t1=t1, w0=w0, tol=1e-5)
      bt_py_time = _time_query(bm, ts)
      logging.warning(
          f"{msg} (torchsde.BrownianTree): {bt_py_time:.4f}"
      )
  
      bm = torchsde.BrownianInterval(
          t0=t0, t1=t1, size=w0.shape, dtype=w0.dtype, device=w0.device
      )
      bi_py_time = _time_query(bm, ts)
      logging.warning(
          f"{msg} (torchsde.BrownianInterval): {bi_py_time:.4f}"
      )
  
      return bp_py_time, bt_py_time, bi_py_time
  
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 2 for calling a global function with example usage: _compare(w0, ts, msg) and returns: bp_py_time,  bt_py_time,  bi_py_time