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,