Skip to content

Commit 04883f4

Browse files
cavenesstf-data-validation-team
caveness
authored and
tf-data-validation-team
committed
no-op
PiperOrigin-RevId: 271424009
1 parent e3b08cf commit 04883f4

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

tensorflow_data_validation/tools/windows/pip/build_tfdv_windows.sh

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,28 @@ pip uninstall -y Cython
139139
pip install dist/*.whl
140140
pip install ${TENSORFLOW}
141141

142+
# If running with tf-nightly, install TFT at head. If installing TFT at head,
143+
# also install TFX-BSL at head.
144+
if [[ ${TENSORFLOW}==tf-nightly ]]; then
145+
echo "Installing TFX-BSL at head"
146+
pushd tfx_bsl_at_head
147+
PYARROW_REQUIREMENT=$(python -c "fp = open('third_party/pyarrow_version.bzl', 'r'); d = {}; exec(fp.read(), d); fp.close(); print(d['PY_DEP'])")
148+
pip install "${PYARROW_REQUIREMENT}"
149+
./configure.sh
150+
bazel run -c opt --copt=-DWIN32_LEAN_AND_MEAN tfx_bsl:build_pip_package -- --python_bin_path ${PYTHON_BIN_PATH}
151+
BSL_WHEEL_PATH=$(find dist -name "*.whl")
152+
pip install ${BSL_WHEEL_PATH}
153+
popd # pop tfx_bsl_at_head
154+
155+
pip uninstall -y tensorflow-transform
156+
echo "Installing TFT at head"
157+
pushd tft_at_head
158+
"${PYTHON_BIN_PATH}" setup.py bdist_wheel
159+
TFT_WHEEL_PATH=$(find dist -name "tensorflow_transform-*.whl")
160+
pip install ${TFT_WHEEL_PATH}
161+
popd # pop tft_at_head
162+
fi
163+
142164
run_py_tests "tensorflow_data_validation" $@
143165

144166
# copy wheel to ${KOKORO_ARTIFACTS_DIR}

0 commit comments

Comments
 (0)