From e4a933e875f25dcd8e0dcbbf8c0279cb516ee3da Mon Sep 17 00:00:00 2001
From: Adam Cogdell <adamcogdell@google.com>
Date: Fri, 1 Nov 2024 13:02:49 -0700
Subject: [PATCH] Reverts df439c009aaac93ef17b7f2395d0eb735c111d23

PiperOrigin-RevId: 692267805
---
 tensorflow/python/saved_model/BUILD           | 32 ++++++++++++++++++-
 .../saved_model/pywrap_saved_model_metrics.cc |  2 ++
 2 files changed, 33 insertions(+), 1 deletion(-)

diff --git a/tensorflow/python/saved_model/BUILD b/tensorflow/python/saved_model/BUILD
index ad65ab126ea5b4..9cac1eff555aec 100644
--- a/tensorflow/python/saved_model/BUILD
+++ b/tensorflow/python/saved_model/BUILD
@@ -2,7 +2,7 @@
 # TensorFlow SavedModel.
 
 load("//tensorflow:strict.default.bzl", "py_strict_library")
-load("//tensorflow:tensorflow.bzl", "if_google")
+load("//tensorflow:tensorflow.bzl", "if_google", "tf_cc_test")
 load("//tensorflow:tensorflow.default.bzl", "cuda_py_strict_test", "tf_py_strict_test", "tf_pybind_cc_library_wrapper", "tf_python_pybind_extension")
 
 package(
@@ -1075,3 +1075,33 @@ tf_py_strict_test(
         "//tensorflow/python/trackable:base",
     ],
 )
+
+# copybara:uncomment_begin(google-only)
+#
+# tf_cc_test(
+#     name = "lineage_logging_test",
+#     srcs = ["lineage_logging_test.cc"],
+#     deps = [
+#         "//learning/brain/contrib/hub/public:client",
+#         "//learning/brain/contrib/hub/public:handle",
+#         "//learning/brain/contrib/hub/public/testing:in_process_tfhub_env",
+#         "//learning/brain/contrib/hub/public/testing:testdata_util",
+#         "//learning/metadata/cataline/proto:ml_data_processes_cc_proto",
+#         "//learning/metadata/lineage_log/cc:lineage_log",
+#         "//learning/metadata/lineage_log/cc:mock_process_client",
+#         "//learning/metadata/lineage_log/cc:test_util",
+#         "//tensorflow/core/platform:protobuf",
+#         "//tensorflow/core/protobuf:for_core_protos_cc",
+#         "@local_tsl//tsl/platform:env",
+#         "@local_tsl//tsl/platform:path",
+#         "//third_party/protobuf",
+#         "@com_google_absl//absl/log:check",
+#         "@com_google_absl//absl/memory",
+#         "@com_google_absl//absl/status",
+#         "@com_google_absl//absl/strings",
+#         "@com_google_absl//absl/strings:string_view",
+#         "@com_google_googletest//:gtest_main",
+#     ],
+# )
+#
+# copybara:uncomment_end
diff --git a/tensorflow/python/saved_model/pywrap_saved_model_metrics.cc b/tensorflow/python/saved_model/pywrap_saved_model_metrics.cc
index 15f15a904a0ab5..a5a3e68c7f7538 100644
--- a/tensorflow/python/saved_model/pywrap_saved_model_metrics.cc
+++ b/tensorflow/python/saved_model/pywrap_saved_model_metrics.cc
@@ -20,6 +20,7 @@ limitations under the License.
 #include <string>
 #include <utility>
 
+// Placeholder for lineage logging import.
 #include "absl/log/log.h"
 #include "absl/status/status.h"
 #include "absl/strings/string_view.h"
@@ -237,6 +238,7 @@ void DefineMetricsModule(py::module main_module) {
         }
         metrics::SavedModelWritePathAndSingleprint().Set(
             path_and_singleprint.value());
+        // Placeholder for lineage logging call.
       },
       py::kw_only(), py::arg("path"), py::arg("singleprint"),
       py::doc("Set the "