Copy all built artifact to pip package

PiperOrigin-RevId: 708016476
tensorflow · Dec 19, 2024 · a81c671 · a81c671
1 parent ddc6b5a
commit a81c671
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugin/build_pip_package.sh b/plugin/build_pip_package.sh
@@ -45,7 +45,7 @@ find tensorboard_plugin_profile/protobuf -name \*.py -exec $sedi -e '
     s/^from plugin.tensorboard_plugin_profile/from tensorboard_plugin_profile/
   ' {} +
 
-cp ${BUILD_WORKSPACE_DIRECTORY}/bazel-bin/external/org_tensorflow/tensorflow/python/profiler/internal/_pywrap_profiler_plugin.so tensorboard_plugin_profile/convert/
+cp ${BUILD_WORKSPACE_DIRECTORY}/bazel-bin/external/org_tensorflow/tensorflow/python/profiler/internal/_pywrap_profiler_plugin.* tensorboard_plugin_profile/convert/
 
 # Copy static files.
 cd tensorboard_plugin_profile