Fix audit wheel issue

PiperOrigin-RevId: 636362500
tensorflow · May 23, 2024 · 34f5e2f · 34f5e2f
1 parent 7d3c6c1
commit 34f5e2f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tensorflow/tools/pip_package/build_pip_package.py b/tensorflow/tools/pip_package/build_pip_package.py
@@ -230,6 +230,10 @@ def patch_so(srcs_dir: str) -> None:
           "tensorflow/compiler/mlir/quantization/tensorflow/python/"
           "pywrap_quantize_model.so"
       ): "$ORIGIN/../../../../../python",
+      (
+          "tensorflow/compiler/mlir/tensorflow_to_stablehlo/python/"
+          "pywrap_tensorflow_to_stablehlo.so"
+      ): "$ORIGIN/../../../../python",
   }
   for file, path in to_patch.items():
     rpath = subprocess.check_output(