File tree 1 file changed +22
-0
lines changed
tensorflow_data_validation/tools/windows/pip 1 file changed +22
-0
lines changed Original file line number Diff line number Diff line change @@ -139,6 +139,28 @@ pip uninstall -y Cython
139
139
pip install dist/* .whl
140
140
pip install ${TENSORFLOW}
141
141
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
+
142
164
run_py_tests " tensorflow_data_validation" $@
143
165
144
166
# copy wheel to ${KOKORO_ARTIFACTS_DIR}
You can’t perform that action at this time.
0 commit comments