chore: add python to CI for tests

huggingface · Jan 21, 2024 · 8c2b54d · 8c2b54d
1 parent 0afb27d
commit 8c2b54d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/.github/workflows/rust.yml b/.github/workflows/rust.yml
@@ -33,6 +33,12 @@ jobs:
           sudo apt-get update
           sudo apt install -y libegl1-mesa libgl1-mesa-dri libxcb-xfixes0-dev vulkan-sdk mesa-vulkan-drivers pkg-config libasound2-dev
 
+      - name: Setup python 
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.10.6'
+          cache: 'pip'
+      - run: pip install -r requirements.txt
       - name: Setup
         run: |
           cargo install wasm-pack

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,3 @@
+--extra-index-url https://download.pytorch.org/whl/cpu
+numpy==1.24.3
+torch==2.0.1