diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml index b734f15d..01ccfebd 100644 --- a/.github/workflows/python-package.yml +++ b/.github/workflows/python-package.yml @@ -32,10 +32,7 @@ jobs: run: | python -m pip install --upgrade pip pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu - pip install -e ".[test,build,yuanrong]" - # Pin below 0.3.12: the 0.3.12 non-cuda wheel ships a mooncake_master binary - # linked against libcudart.so.12, which fails to start on runners without CUDA. - pip install "mooncake-transfer-engine-non-cuda<0.3.12" + pip install -e ".[test,build,yuanrong,mooncake]" - name: Lint with flake8 run: | # stop the build if there are Python syntax errors or undefined names diff --git a/.github/workflows/run-tests.yml b/.github/workflows/run-tests.yml index 05bba3ae..e08358b7 100644 --- a/.github/workflows/run-tests.yml +++ b/.github/workflows/run-tests.yml @@ -33,9 +33,7 @@ jobs: python -m pip install --upgrade pip pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu pip install -e ".[test,build,yuanrong]" - # Pin below 0.3.12: the 0.3.12 non-cuda wheel ships a mooncake_master binary - # linked against libcudart.so.12, which fails to start on runners without CUDA. - pip install "mooncake-transfer-engine-non-cuda<0.3.12" + pip install "mooncake-transfer-engine-non-cuda" - name: Run Tests run: | pytest tests