Skip to content

Commit f4d56c6

Browse files
amirafzalifacebook-github-bot
authored andcommitted
manylinux support for nightly wheel (#416)
Summary: Rollback Plan: Differential Revision: D77689672
1 parent d07cbce commit f4d56c6

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

.github/workflows/wheels.yml

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,11 @@ jobs:
4343
4444
python setup.py bdist_wheel
4545
46-
# hacky until the right distribution wheel can be made...
47-
find dist -name "*linux_x86_64.whl" -type f -exec bash -c 'mv "$1" "${1/linux_x86_64.whl/manylinux2014_x86_64.whl}"' _ {} \;
46+
# Repair wheel (distribution support)
47+
pip install auditwheel
48+
auditwheel -v repair dist/* --plat manylinux2014_x86_64 --exclude libtorch_python.so --exclude libc10.so --exclude libtorch.so --exclude libtorch_cpu.so
49+
rm -rf dist/*
50+
mv wheelhouse/* dist/
4851
ls -la dist/
4952
5053
# Run tests

python/tests/tools/test_commands.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -230,7 +230,7 @@ async def test_get_or_create(self) -> None:
230230
check_interval=_5_MS,
231231
)
232232

233-
mock_create.called_once_with(config, appdef)
233+
mock_create.assert_called_once_with(config, appdef)
234234
self.assertEqual(server_info.server_handle, "slurm:///456")
235235
self.assertListEqual(
236236
mock_info.call_args_list,

0 commit comments

Comments
 (0)