code: def memory_usage_psutil():
      
      process = psutil.Process(os.getpid())
      mem = process.memory_info()[0] / float(2 ** 20)
      mem_vms = process.memory_info()[1] / float(2 ** 20)
      return mem, mem_vms
docs: Return the current process memory usage in MB.