Skip to content

Commit 9dc31f9

Browse files
authored
[Cherry-pick] More packaging (#5) (#6)
1 parent 3333893 commit 9dc31f9

File tree

3 files changed

+11
-0
lines changed

3 files changed

+11
-0
lines changed

.github/workflows/linux_wheel.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,5 +50,6 @@ jobs:
5050
test-infra-ref: main
5151
build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
5252
pre-script: packaging/pre_build_script.sh
53+
post-script: packaging/post_build_script.sh
5354
package-name: torchvision_extra_decoders
5455
trigger-event: ${{ github.event_name }}

packaging/post_build_script.sh

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
#!/bin/bash
2+
3+
set -ex
4+
5+
old="linux_x86_64"
6+
new="manylinux_2_17_x86_64.manylinux2014_x86_64"
7+
echo "Replacing ${old} with ${new} in wheel name"
8+
mv dist/*${old}*.whl $(echo dist/*${old}*.whl | sed "s/${old}/${new}/")

setup.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,8 @@ def get_and_write_version():
3535
elif sha != "Unknown":
3636
version += "+" + sha[:7]
3737

38+
version = version.replace("+cpu", "")
39+
3840
with open(ROOT_DIR / "torchvision_extra_decoders/version.py", "w") as f:
3941
f.write(f"__version__ = '{version}'\n")
4042
f.write(f"git_version = {repr(sha)}\n")

0 commit comments

Comments
 (0)