code: def is_gke_env():
      return os.environ.get("TPU_WORKER_HOSTNAMES", None) is not None
  
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 5 for calling a global function with example usage: is_gke_env() and returns: unknown