code: def is_gce_env():
      worker_number_string = get_metadata("agent-worker-number")
      try:
          worker_number = int(worker_number_string)
          return True
      except:
          return False
  
imports: from __future__ import annotations
  import os
  import re
  import socket
  import time
  from jax._src import clusters
  from jax._src.cloud_tpu_init import running_in_cloud_tpu_vm
label_desc: Use image node_id 3 for calling a global function with example usage: is_gce_env() and returns: True, False