code: def mpi_avg(x):
      """Average a scalar or vector over MPI processes."""
      return mpi_sum(x) / num_procs()
  
file_path: spinningup/spinup/utils/mpi_tools.py
imports: from mpi4py import MPI
  import os, subprocess, sys
  import numpy
label_desc: Use image node_id 9 for calling a global function with example usage: mpi_avg(x) and returns: unknown