Split source package and wheel build.

tensorflower-gardener · tensorflower-gardener · commit a0fb6b0a517d · 2025-09-04T12:40:33.000-07:00
PiperOrigin-RevId: 803134650
diff --git a/tf_keras/wheel/wheel.bzl b/tf_keras/wheel/wheel.bzl
@@ -65,6 +65,8 @@ def _wheel_impl(ctx):
         )
         args.add("--build-source-package-only", "True")
     output_file = ctx.actions.declare_file(output_path + "/" + name)
+
+    # TODO(ybaturina): Make this directory name dynamic to avoid collisions.
     copied_whl_srcs_dir = ctx.actions.declare_directory("copied_whl_srcs")
     outputs = [copied_whl_srcs_dir, output_file]
     args.add("--whl_dir", output_file.path[:output_file.path.rfind("/")])

Original file line number	Diff line number	Diff line change
`@@ -65,6 +65,8 @@ def _wheel_impl(ctx):`
`65`	`65`	`)`
`66`	`66`	`args.add("--build-source-package-only", "True")`
`67`	`67`	`output_file = ctx.actions.declare_file(output_path + "/" + name)`
	`68`	`+`
	`69`	`+ # TODO(ybaturina): Make this directory name dynamic to avoid collisions.`
`68`	`70`	`copied_whl_srcs_dir = ctx.actions.declare_directory("copied_whl_srcs")`
`69`	`71`	`outputs = [copied_whl_srcs_dir, output_file]`
`70`	`72`	`args.add("--whl_dir", output_file.path[:output_file.path.rfind("/")])`