[one-cmds] Revise one-prepare-venv.u2204 (Samsung#10779)

This will revise one-prepare-venv.u2204 to match tensorflow addon packages version. ONE-DCO-1.0-Signed-off-by: SaeHie Park <[email protected]>
wooksong · May 12, 2023 · ad3ee22 · ad3ee22
1 parent bf09309
commit ad3ee22
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/compiler/one-cmds/one-prepare-venv.u2204 b/compiler/one-cmds/one-prepare-venv.u2204
@@ -64,7 +64,9 @@ else
   ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install tensorflow-cpu==${VER_TENSORFLOW}
 fi
 ${VENV_PYTHON} -m pip ${PIP_OPTIONS} install Pillow
-${VENV_PYTHON} -m pip ${PIP_OPTIONS} install tensorflow_probability
+# https://github.com/Samsung/ONE/issues/10777
+${VENV_PYTHON} -m pip ${PIP_OPTIONS} install tensorflow_probability==0.18.0
+${VENV_PYTHON} -m pip ${PIP_OPTIONS} install tensorflow_addons==0.18.0
 
 # Install PyTorch and ONNX related
 # NOTE set ONE_PREPVENV_TORCH_STABLE to override 'torch_stable.html' URL.