use newer cluster version

pyg-team · Jun 29, 2019 · f5a1f89 · f5a1f89
1 parent 9835dcf
commit f5a1f89
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -18,7 +18,7 @@ before_install:
 install:
   - pip install numpy
   - pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
-  - pip install torch-scatter torch-sparse torch-cluster torch-spline-conv
+  - pip install torch-scatter torch-sparse torch-cluster==1.4.3a1 torch-spline-conv
   - pip install cython && pip install gdist
   - pip install pycodestyle
   - pip install flake8

diff --git a/docs/requirements_2.txt b/docs/requirements_2.txt
@@ -1,3 +1,3 @@
 torch-scatter
 torch-sparse
-torch-cluster
+torch-cluster==1.4.3a1