[CI] Install typing-extensions before installing pytorch pre

Dao-AILab · Jan 30, 2024 · dc84790 · dc84790
1 parent de71db2
commit dc84790
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -126,6 +126,7 @@ jobs:
           export TORCH_CUDA_VERSION=$(python -c "import os; minv = {'1.12': 113, '1.13': 116, '2.0': 117, '2.1': 118, '2.2': 118}[os.environ['MATRIX_TORCH_VERSION']]; maxv = {'1.12': 116, '1.13': 117, '2.0': 118, '2.1': 121, '2.2': 121}[os.environ['MATRIX_TORCH_VERSION']]; print(max(min(int(os.environ['MATRIX_CUDA_VERSION']), maxv), minv))")
           if [[ ${{ matrix.torch-version }} == *"dev"* ]]; then
             # --no-deps because we can't install old versions of pytorch-triton
+            pip install typing-extensions jinja2
             pip install --no-cache-dir --no-deps --pre https://download.pytorch.org/whl/nightly/cu${TORCH_CUDA_VERSION}/torch-${{ matrix.torch-version }}%2Bcu${TORCH_CUDA_VERSION}-cp${MATRIX_PYTHON_VERSION}-cp${MATRIX_PYTHON_VERSION}-linux_x86_64.whl
           else
             pip install --no-cache-dir torch==${{ matrix.torch-version }} --index-url https://download.pytorch.org/whl/cu${TORCH_CUDA_VERSION}

diff --git a/fast_hadamard_transform/__init__.py b/fast_hadamard_transform/__init__.py
@@ -1,3 +1,3 @@
-__version__ = "1.0.3.post2"
+__version__ = "1.0.3.post3"
 
 from fast_hadamard_transform.fast_hadamard_transform_interface import hadamard_transform