Move last remaining use of loader_impl._parse_saved_model to loader_i…

…mpl.parse_saved_model PiperOrigin-RevId: 412912759
tensorflow · Nov 29, 2021 · e40893d · e40893d
1 parent 5f53a7d
commit e40893d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_estimator/python/estimator/estimator_test.py b/tensorflow_estimator/python/estimator/estimator_test.py
@@ -3338,7 +3338,7 @@ def _validate_strip_default_attrs(self, estimator_cls, export_fn,
         tf.compat.as_bytes(tmpdir), tf.compat.as_bytes('export'))
 
     export_dir = export_fn(est, export_dir_base, serving_input_receiver_fn)
-    saved_model_pb = loader_impl._parse_saved_model(export_dir)
+    saved_model_pb = loader_impl.parse_saved_model(export_dir)
     self.assertIsNotNone(saved_model_pb)
     meta_graph_def = [
         x for x in saved_model_pb.meta_graphs