Skip to content

Commit 04d8fc4

Browse files
NicolasHugjithunnair-amd
authored andcommitted
Fix post-build-script (pytorch#8982)
(cherry picked from commit 29066f5)
1 parent 3ac97aa commit 04d8fc4

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

packaging/pre_build_script.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ else
3333
fi
3434

3535
yum install -y libjpeg-turbo-devel libwebp-devel freetype gnutls
36-
pip install auditwheel
36+
pip install "auditwheel<6.3.0"
3737
fi
3838

3939
pip install numpy pyyaml future ninja

packaging/wheel/relocate.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,10 @@
1515

1616
# Third party imports
1717
if sys.platform == "linux":
18-
from auditwheel.lddtree import lddtree
18+
try:
19+
from auditwheel.lddtree import lddtree
20+
except ImportError:
21+
from auditwheel import lddtree
1922

2023

2124
ALLOWLIST = {

0 commit comments

Comments
 (0)