Fix git action.

.github/workflows/docs.yml

@@ -26,7 +26,8 @@ jobs:
       # ADJUST THIS: install all dependencies (including pdoc)
       - name: Install dependencies
         run: |
-          pip install torch
+          pip install torch==2.1.0
+          pip install torch-scatter -f https://data.pyg.org/whl/torch-2.1.0+cpu.html
           pip install -e .
       # ADJUST THIS: build your documentation into docs/.
       # We use a custom build script for pdoc itself, ideally you just run `pdoc -o docs/ ...` here.