Use pyrwap rules for LiteRT repo to fix duplicate registration errors

PiperOrigin-RevId: 758995263

Author: ecalubaquib

tflite/python/BUILD

@@ -1,7 +1,8 @@
 load("@flatbuffers//:build_defs.bzl", "flatbuffer_py_library")
 load("@org_tensorflow//tensorflow:pytype.default.bzl", "pytype_strict_contrib_test", "pytype_strict_library")
 load("@org_tensorflow//tensorflow:strict.default.bzl", "py_strict_binary", "py_strict_library", "py_strict_test")
-load("@org_tensorflow//tensorflow:tensorflow.default.bzl", "get_compatible_with_portable")
+load("@org_tensorflow//tensorflow:tensorflow.default.bzl", "get_compatible_with_portable", "pywrap_library")
+load("@org_tensorflow//tensorflow/core/platform:build_config_root.bzl", "if_pywrap")
 load("@rules_shell//shell:sh_test.bzl", "sh_test")
 load("//tflite:special_rules.bzl", "internal_visibility_allowlist")
 
@@ -633,3 +634,46 @@ py_strict_test(
         "@org_tensorflow//tensorflow/python/trackable:autotrackable",
     ],
 )
+
+pywrap_library(
+    name = "pywrap_tflite",
+    common_lib_linkopts = {
+        "framework": select({
+            "@bazel_tools//src/conditions:windows": [],
+            "@bazel_tools//src/conditions:darwin": [],
+            "//conditions:default": ["-z defs"],
+        }),
+    },
+    common_lib_version_scripts = {
+        "framework": select({
+            "@bazel_tools//src/conditions:windows": None,
+            "@bazel_tools//src/conditions:darwin": ":darwin_framework.lds",
+            "//conditions:default": ":linux_framework.lds",
+        }),
+    },
+    common_lib_versions = {
+        "framework": "2",
+    },
+    pywrap_count = 4,
+    visibility = ["//visibility:public"],
+    deps = [
+        "//tflite/python/analyzer_wrapper:_pywrap_analyzer_wrapper",
+        "//tflite/python/interpreter_wrapper:_pywrap_tensorflow_interpreter_wrapper",
+        "//tflite/python/metrics:_pywrap_tensorflow_lite_metrics_wrapper",
+        "//tflite/python/optimize:_pywrap_tensorflow_lite_calibration_wrapper",
+    ],
+)
+
+py_strict_test(
+    name = "my_test",
+    srcs = ["my_test.py"],
+    visibility = ["//visibility:public"],
+    deps = [
+        "//tflite/python:lite",
+        "@org_tensorflow//tensorflow:tensorflow_py",
+    ] + if_pywrap(
+        if_true = [
+            ":pywrap_tflite",
+        ],
+    ),
+)

tflite/python/my_test.py

@@ -0,0 +1,58 @@
+# Copyright 2021 The TensorFlow Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+"""Tests for mytest package."""
+
+
+import logging
+import traceback
+import unittest
+
+# pylint: disable=g-import-not-at-top
+try:
+  import tensorflow as tf
+except ImportError as e:
+  logging.exception("Error during import: %s", e)
+  logging.exception("\nImport stack trace:")
+  traceback.print_exc()
+  raise  # Re-raise the exception so the test runner knows it failed
+except Exception as e:
+  logging.exception("A different error occurred during import: %s", e)
+  logging.exception("\nImport stack trace:")
+  traceback.print_exc()
+  raise
+
+try:
+  from tflite.python import lite
+except ImportError as e:
+  logging.exception("Error during import: %s", e)
+  logging.exception("\nImport stack trace:")
+  traceback.print_exc()
+  raise  # Re-raise the exception so the test runner knows it failed
+except Exception as e:
+  logging.exception("A different error occurred during import: %s", e)
+  logging.exception("\nImport stack trace:")
+  traceback.print_exc()
+  raise
+# pylint: enable=g-import-not-at-top
+
+
+class TestArithmeticFunctions(unittest.TestCase):
+  def test_add_positive_numbers(self):
+    logging.info("hello")
+    logging.info(tf.__version__)
+    logging.info(lite)
+
+if __name__ == "__main__":
+  unittest.main()

tflite/testing/BUILD

@@ -760,6 +760,8 @@ edgetpu_ops = [
     "transpose_conv",
 ]
 
+# THIS IS A TEST
+
 # copybara:uncomment_begin(google-only)
 # [gen_zipped_test_file(
 #     name = "zip_%s_edgetpu" % op_name,

tflite/testing/string_util_wrapper.cc

@@ -12,7 +12,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
 ==============================================================================*/
-
+/* THIS IS A TEST COMMENT*/
 #include "pybind11/pybind11.h"  // from @pybind11
 #include "pybind11/pytypes.h"  // from @pybind11
 #include "tflite/python/interpreter_wrapper/numpy.h"