code: def _gen_tfdv_schema(stats, args):
      schema = tfdv.infer_schema(stats)
      _save_tfdv_schema(schema, args)
      return schema
  
file_path: guildai/examples/bias/gen_schema.py
imports: import argparse
  import os
  import tensorflow_data_validation
  import tensorflow_data_validation.utils.display_util
label_desc: Use image node_id 6 for calling a global function with example usage: _gen_tfdv_schema(stats, args) and returns: schema