Updated to use tf.io.gfile for file operations.

PiperOrigin-RevId: 276330599

Author: mdreves

RELEASE.md

@@ -4,6 +4,9 @@
 
 ## Bug fixes and other changes
 
+*   Updated to use tf.io.gfile for reading config files (fixes issue with
+    reading from GCS/HDFS in 0.15.0 and 0.15.1 releases).
+
 ## Breaking changes
 
 ## Deprecations

tensorflow_model_analysis/api/model_eval_lib.py

@@ -85,8 +85,8 @@ def _serialize_eval_config(eval_config: config.EvalConfig) -> Text:
 def load_eval_config(output_path: Text) -> config.EvalConfig:
   """Loads eval config."""
   path = os.path.join(output_path, _EVAL_CONFIG_FILE)
-  if os.path.exists(path):
-    with open(path) as f:
+  if tf.io.gfile.exists(path):
+    with tf.io.gfile.GFile(path, 'r') as f:
       pb = json_format.Parse(f.read(), config_pb2.EvalConfigAndVersion())
       _check_version(pb.version, output_path)
       return pb.eval_config

tensorflow_model_analysis/api/model_eval_lib_test.py

@@ -748,7 +748,8 @@ def testSerializeDeserializeEvalConfig(self):
                 })
         ],
         options=options)
-    with open(os.path.join(output_path, 'eval_config.json'), 'w') as f:
+    with tf.io.gfile.GFile(os.path.join(output_path, 'eval_config.json'),
+                           'w') as f:
       f.write(model_eval_lib._serialize_eval_config(eval_config))
     got_eval_config = model_eval_lib.load_eval_config(output_path)
     self.assertEqual(eval_config, got_eval_config)