File tree Expand file tree Collapse file tree 1 file changed +12
-1
lines changed
Expand file tree Collapse file tree 1 file changed +12
-1
lines changed Original file line number Diff line number Diff line change @@ -3,16 +3,27 @@ block_cipher = None
33
44from glob import iglob
55from os .path import basename , dirname , abspath
6+ import os
7+ import fnmatch
68
79script_name = '%%SCRIPT%%'
810train_libs = % % TRAIN_LIBS % %
911strip = True
1012site_packages = '.venv/lib/python3.6/site-packages/'
1113
14+
15+ def recursive_glob (treeroot , pattern ):
16+ results = []
17+ for base , dirs , files in os .walk (treeroot ):
18+ goodfiles = fnmatch .filter (files , pattern )
19+ results .extend (os .path .join (base , f ) for f in goodfiles )
20+ return results
21+
22+
1223if train_libs :
1324 binaries = [
1425 (abspath (i ), dirname (i .replace (site_packages , '' )))
15- for i in iglob (site_packages + "tensorflow/**/* .so" , recursive = True )
26+ for i in recursive_glob (site_packages + "tensorflow/" , "* .so" )
1627 ]
1728else :
1829 binaries = []
You can’t perform that action at this time.
0 commit comments