Move write_json_file to file_util.

PiperOrigin-RevId: 348550288

Author: lintian06

tensorflow_examples/lite/model_maker/core/data_util/text_dataloader.py

@@ -19,7 +19,6 @@
 
 import csv
 import hashlib
-import json
 import os
 import random
 import tempfile
@@ -69,12 +68,6 @@ def _get_cache_filenames(cache_dir, model_spec, data_name, is_training):
   return cache_tfrecord_file, cache_meta_data_file, cache_prefix
 
 
-def _write_meta_data(meta_data_file, meta_data):
-  """Writes meta data into file."""
-  with tf.io.gfile.GFile(meta_data_file, 'w') as f:
-    json.dump(meta_data, f)
-
-
 def _get_cache_info(cache_dir, data_name, model_spec, is_training):
   """Gets cache related information: whether is cached, related filenames."""
   if cache_dir is None:
@@ -259,7 +252,7 @@ def _save_data(cls, examples, model_spec, label_names, tfrecord_file,
         'num_classes': len(label_names),
         'index_to_label': label_names
     }
-    _write_meta_data(meta_data_file, meta_data)
+    file_util.write_json_file(meta_data_file, meta_data)
 
   @classmethod
   def _get_cache_info(cls, cache_dir, data_name, model_spec, is_training):
@@ -337,7 +330,7 @@ def from_squad(cls,
         filename, model_spec, tfrecord_file, is_training,
         version_2_with_negative)
 
-    _write_meta_data(meta_data_file, meta_data)
+    file_util.write_json_file(meta_data_file, meta_data)
 
     dataset, meta_data = _load(tfrecord_file, meta_data_file, model_spec,
                                is_training)

tensorflow_examples/lite/model_maker/core/file_util.py

@@ -23,5 +23,12 @@
 
 
 def load_json_file(json_file):
+  """Loads json data from file."""
   with tf.io.gfile.GFile(json_file, 'r') as reader:
     return json.load(reader)
+
+
+def write_json_file(json_file, data):
+  """Writes json data into file."""
+  with tf.io.gfile.GFile(json_file, 'w') as f:
+    json.dump(data, f)