From 951e0468d1802bcf37d3003b1ea74f3b4ae5d386 Mon Sep 17 00:00:00 2001 From: Madhur Karampudi <142544288+vkarampudi@users.noreply.github.com> Date: Sun, 17 May 2026 15:55:47 -0700 Subject: [PATCH 1/6] Update setup.py --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index a6645298..7c3c6205 100644 --- a/setup.py +++ b/setup.py @@ -219,13 +219,13 @@ def select_constraint(default, nightly=None, git_master=None): "tensorflow>=2.21,<2.22", "tensorflow-metadata" + select_constraint( - default=">=1.17.1,<1.18", + default="@git+https://github.com/tensorflow/metadata@master", nightly=">=1.18.0.dev", git_master="@git+https://github.com/tensorflow/metadata@master", ), "tfx-bsl" + select_constraint( - default=">=1.17.1,<1.18", + default="@git+https://github.com/tensorflow/tfx-bsl@master", nightly=">=1.18.0.dev", git_master="@git+https://github.com/tensorflow/tfx-bsl@master", ), From 811f1d81892ebf014c9b1da7a6609b6f690ac059 Mon Sep 17 00:00:00 2001 From: Madhur Karampudi <142544288+vkarampudi@users.noreply.github.com> Date: Sun, 17 May 2026 16:49:27 -0700 Subject: [PATCH 2/6] Update setup.py --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 7c3c6205..7cf50c04 100644 --- a/setup.py +++ b/setup.py @@ -225,7 +225,7 @@ def select_constraint(default, nightly=None, git_master=None): ), "tfx-bsl" + select_constraint( - default="@git+https://github.com/tensorflow/tfx-bsl@master", + default="@git+https://github.com/tensorflow/tfx-bsl@testing", nightly=">=1.18.0.dev", git_master="@git+https://github.com/tensorflow/tfx-bsl@master", ), From f495fc82c7fdcaeeb74af0b7db40cfc320c9c038 Mon Sep 17 00:00:00 2001 From: Venkata Sai Madhur Karampudi Date: Sun, 17 May 2026 23:52:37 +0000 Subject: [PATCH 3/6] Point tfx-bsl dependency to vkarampudi/tfx-bsl@testing fork --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 7cf50c04..97858ef9 100644 --- a/setup.py +++ b/setup.py @@ -225,7 +225,7 @@ def select_constraint(default, nightly=None, git_master=None): ), "tfx-bsl" + select_constraint( - default="@git+https://github.com/tensorflow/tfx-bsl@testing", + default="@git+https://github.com/vkarampudi/tfx-bsl@testing", nightly=">=1.18.0.dev", git_master="@git+https://github.com/tensorflow/tfx-bsl@master", ), From f2134125d95b466eb3b31a96281702ffaa12178c Mon Sep 17 00:00:00 2001 From: Venkata Sai Madhur Karampudi Date: Mon, 18 May 2026 18:23:28 +0000 Subject: [PATCH 4/6] Widen ipython dependency to <9 in visualization requirements for Python 3.13 compatibility --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 97858ef9..fcbcffae 100644 --- a/setup.py +++ b/setup.py @@ -126,7 +126,7 @@ def _make_mutual_information_requirements(): def _make_visualization_requirements(): return [ - "ipython>=7,<8", + "ipython>=7,<9", ] From e831327cbf2d5cba070de7e69042aea1c7fc0215 Mon Sep 17 00:00:00 2001 From: Venkata Sai Madhur Karampudi Date: Mon, 1 Jun 2026 21:20:18 +0000 Subject: [PATCH 5/6] Fix Bazel PEP 517 Build Environment Isolation and Sandbox Restrictions --- setup.py | 25 +++++++++++++++++++++---- third_party/python_configure.bzl | 7 ++++++- 2 files changed, 27 insertions(+), 5 deletions(-) diff --git a/setup.py b/setup.py index fcbcffae..19cc6405 100644 --- a/setup.py +++ b/setup.py @@ -90,10 +90,27 @@ def run(self): bazel_version = f.read().strip() os.environ["USE_BAZEL_VERSION"] = bazel_version + pythonpath = os.environ.get("PYTHONPATH", "") + # Include sys.path entries to support PEP 517 isolated build environments + sys_path_entries = os.pathsep.join([p for p in sys.path if p]) + if pythonpath: + pythonpath = os.pathsep.join([pythonpath, sys_path_entries]) + else: + pythonpath = sys_path_entries + + bazel_args = [ + self._bazel_cmd, + "run", + "-c", + "opt", + f"--repo_env=PYTHON_BIN_PATH={sys.executable}", + f"--repo_env=PYTHONPATH={pythonpath}", + ] + bazel_args.extend(self._additional_build_options) + bazel_args.append("//tensorflow_data_validation:move_generated_files") + subprocess.check_call( - [self._bazel_cmd, "run", "-c", "opt"] - + self._additional_build_options - + ["//tensorflow_data_validation:move_generated_files"], + bazel_args, # Bazel should be invoked in a directory containing bazel WORKSPACE # file, which is the root directory. cwd=os.path.dirname(os.path.realpath(__file__)), @@ -225,7 +242,7 @@ def select_constraint(default, nightly=None, git_master=None): ), "tfx-bsl" + select_constraint( - default="@git+https://github.com/vkarampudi/tfx-bsl@testing", + default="@git+https://github.com/vkarampudi/tfx-bsl@master", nightly=">=1.18.0.dev", git_master="@git+https://github.com/tensorflow/tfx-bsl@master", ), diff --git a/third_party/python_configure.bzl b/third_party/python_configure.bzl index b4980f73..c7ec08e0 100644 --- a/third_party/python_configure.bzl +++ b/third_party/python_configure.bzl @@ -205,7 +205,11 @@ def _raw_exec(repository_ctx, cmdline): Returns: The 'exec_result' of repository_ctx.execute(). """ - return repository_ctx.execute(cmdline) + env = {} + for k in ["PYTHONPATH", "PYTHON_BIN_PATH", "PYTHON_LIB_PATH"]: + if k in repository_ctx.os.environ: + env[k] = repository_ctx.os.environ[k] + return repository_ctx.execute(cmdline, environment = env) def _read_dir(repository_ctx, src_dir): """Returns a sorted list with all files in a directory. @@ -476,6 +480,7 @@ _ENVIRONS = [ BAZEL_SH, PYTHON_BIN_PATH, PYTHON_LIB_PATH, + "PYTHONPATH", ] local_python_configure = repository_rule( From 3971cb71cf0c84e2b2f1e15f472d6303faebd3ba Mon Sep 17 00:00:00 2001 From: Venkata Sai Madhur Karampudi Date: Mon, 1 Jun 2026 22:29:44 +0000 Subject: [PATCH 6/6] Point tfx-bsl dependency to official tensorflow/tfx-bsl@master branch --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 19cc6405..e40c7a08 100644 --- a/setup.py +++ b/setup.py @@ -242,7 +242,7 @@ def select_constraint(default, nightly=None, git_master=None): ), "tfx-bsl" + select_constraint( - default="@git+https://github.com/vkarampudi/tfx-bsl@master", + default="@git+https://github.com/tensorflow/tfx-bsl@master", nightly=">=1.18.0.dev", git_master="@git+https://github.com/tensorflow/tfx-bsl@master", ),