From f4d56c6ca3fd1c405e3b464c7db8cda584422673 Mon Sep 17 00:00:00 2001 From: Amir Afzali Date: Wed, 2 Jul 2025 15:09:18 -0700 Subject: [PATCH] manylinux support for nightly wheel (#416) Summary: Rollback Plan: Differential Revision: D77689672 --- .github/workflows/wheels.yml | 7 +++++-- python/tests/tools/test_commands.py | 2 +- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml index f7425565..1d87a103 100644 --- a/.github/workflows/wheels.yml +++ b/.github/workflows/wheels.yml @@ -43,8 +43,11 @@ jobs: python setup.py bdist_wheel - # hacky until the right distribution wheel can be made... - find dist -name "*linux_x86_64.whl" -type f -exec bash -c 'mv "$1" "${1/linux_x86_64.whl/manylinux2014_x86_64.whl}"' _ {} \; + # Repair wheel (distribution support) + pip install auditwheel + auditwheel -v repair dist/* --plat manylinux2014_x86_64 --exclude libtorch_python.so --exclude libc10.so --exclude libtorch.so --exclude libtorch_cpu.so + rm -rf dist/* + mv wheelhouse/* dist/ ls -la dist/ # Run tests diff --git a/python/tests/tools/test_commands.py b/python/tests/tools/test_commands.py index c7aeb776..a2e08d73 100644 --- a/python/tests/tools/test_commands.py +++ b/python/tests/tools/test_commands.py @@ -230,7 +230,7 @@ async def test_get_or_create(self) -> None: check_interval=_5_MS, ) - mock_create.called_once_with(config, appdef) + mock_create.assert_called_once_with(config, appdef) self.assertEqual(server_info.server_handle, "slurm:///456") self.assertListEqual( mock_info.call_args_list,