code: def write_ckpt(self, path, shard):
      self.input_q.put(("write_ckpt", (path, shard)))
      return self.output_q.get()
  
file_path: mesh-transformer-jax/mesh_transformer/train_actor.py
imports: import ray
  import time
  import numpy
  from queue import Queue
  from mesh_transformer.util import head_print
label_desc: Use image node_id 7 for calling the NetworkRunner obj's underlying member method code with example usage: obj.write_ckpt(path, shard) and returns: self