Skip to content

Commit 86317b2

Browse files
prevant to include wrong tensorflow_probability version (#522)
* prevant to include wrong tensorflow_probability version --------- Co-authored-by: yanzhen1233 <wb-yz775491@alibaba-inc.com>
1 parent e33da24 commit 86317b2

File tree

2 files changed

+20
-16
lines changed

2 files changed

+20
-16
lines changed

.gitignore

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,3 +31,6 @@ pai_jobs/easy_rec*.tar.gz
3131
# unit test
3232
/data
3333
/UNIT_TEST_CASE_LIST
34+
35+
.DS_Store
36+
.python-version

pai_jobs/deploy_ext.sh

Lines changed: 17 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -143,26 +143,27 @@ then
143143
rm -rf faiss.tar.gz
144144
fi
145145

146-
if [ ! -d "tensorflow_probability" ]
146+
if [ -d "tensorflow_probability" ]
147147
then
148-
if [ $is_tf15 -gt 0 ]; then
149-
tfp_version='0.8.0'
150-
else
151-
tfp_version='0.5.0'
152-
fi
153-
if [ ! -e "tensorflow_probability" ]
148+
rm -rf tensorflow_probability
149+
fi
150+
if [ $is_tf15 -gt 0 ]; then
151+
tfp_version='0.8.0'
152+
else
153+
tfp_version='0.5.0'
154+
fi
155+
if [ ! -e "tensorflow_probability" ]
156+
then
157+
wget http://easyrec.oss-cn-beijing.aliyuncs.com/3rdparty/probability-${tfp_version}.tar.gz
158+
if [ $? -ne 0 ]
154159
then
155-
wget http://easyrec.oss-cn-beijing.aliyuncs.com/3rdparty/probability-${tfp_version}.tar.gz
156-
if [ $? -ne 0 ]
157-
then
158-
echo "tensorflow_probability download failed."
159-
fi
160+
echo "tensorflow_probability download failed."
160161
fi
161-
tar -xzvf probability-${tfp_version}.tar.gz --strip-components=1 probability-${tfp_version}/tensorflow_probability
162-
rm -rf tensorflow_probability/examples
163-
rm -rf tensorflow_probability/g3doc
164-
rm -rf probability-${tfp_version}.tar.gz
165162
fi
163+
tar -xzvf probability-${tfp_version}.tar.gz --strip-components=1 probability-${tfp_version}/tensorflow_probability
164+
rm -rf tensorflow_probability/examples
165+
rm -rf tensorflow_probability/g3doc
166+
rm -rf probability-${tfp_version}.tar.gz
166167

167168
tar -cvzhf $RES_PATH easy_rec datahub lz4 cprotobuf kafka faiss tensorflow_probability run.py
168169

0 commit comments

Comments
 (0)