Refactor TF build wheel rule.

third_party/tsl/opensource_only.files

@@ -111,6 +111,7 @@ third_party/py/python_init_repositories.bzl:
 third_party/py/python_init_rules.bzl:
 third_party/py/python_init_toolchains.bzl:
 third_party/py/python_repo.bzl:
+third_party/py/python_wheel_version.bzl:
 third_party/pybind11.BUILD:
 third_party/pybind11_bazel/BUILD:
 third_party/python_runtime/BUILD:

third_party/tsl/third_party/py/manylinux_compliance_test.py

@@ -84,7 +84,7 @@ def verify_manylinux_compliance(
   Raises:
     RuntimeError: if the wheel is not manyLinux compliant.
   """
-  regex = 'following platform tag: "{}"'.format(compliance_tag)
+  regex = 'following platform tag:\s+"{}"'.format(compliance_tag)
   if not re.search(regex, auditwheel_log):
     raise RuntimeError(
         ("The wheel is not compliant with the tag {tag}.\n{result}").format(

third_party/tsl/third_party/py/python_wheel_version.bzl

@@ -0,0 +1,24 @@
+""" Repository rule to generate a file with python wheel version. """
+
+def _python_wheel_version_repository_impl(repository_ctx):
+    file_content = repository_ctx.read(
+        repository_ctx.path(repository_ctx.attr.file_with_version),
+    )
+    version_line_start_index = file_content.find(repository_ctx.attr.version_key)
+    version_line_end_index = version_line_start_index + file_content[version_line_start_index:].find("\n")
+    repository_ctx.file(
+        "wheel_version.bzl",
+        file_content[version_line_start_index:version_line_end_index].replace(
+            repository_ctx.attr.version_key,
+            "WHEEL_VERSION",
+        ),
+    )
+    repository_ctx.file("BUILD", "")
+
+python_wheel_version_repository = repository_rule(
+    implementation = _python_wheel_version_repository_impl,
+    attrs = {
+        "file_with_version": attr.label(mandatory = True, allow_single_file = True),
+        "version_key": attr.string(mandatory = True),
+    },
+)