linting fixes

.travis.yml

@@ -10,11 +10,11 @@ install:
   - if [[ $TRAVIS_PYTHON_VERSION == 2.7 ]]; then pip install --quiet http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp27-cp27mu-linux_x86_64.whl; fi
   - if [[ $TRAVIS_PYTHON_VERSION == 3.5 ]]; then pip install --quiet http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp35-cp35m-linux_x86_64.whl; fi
   - if [[ $TRAVIS_PYTHON_VERSION == 3.6 ]]; then pip install --quiet http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp36-cp36m-linux_x86_64.whl; fi
-  - pip install torch-unique
   - pip install pycodestyle
   - pip install flake8
-  - pip install cffi
   - pip install codecov
+  - pip install cffi
+  - pip install torch-unique
 script:
   - pycodestyle .
   - flake8 .

test/test_grid.py

@@ -30,8 +30,9 @@ def test_grid_cluster_cpu(tensor):
 @pytest.mark.parametrize('tensor', tensors)
 def test_grid_cluster_gpu(tensor):  # pragma: no cover
     position = Tensor(tensor, [[0, 0], [9, 9], [2, 8], [2, 2], [8, 3]]).cuda()
-    size = torch.cuda.LongTensor([5, 5])
-    expected = torch.LongTensor([0, 3, 1, 0, 2])
+    print(position)
+    # size = torch.cuda.LongTensor([5, 5])
+    # expected = torch.LongTensor([0, 3, 1, 0, 2])
 
     # output = grid_cluster(position, size)
     # assert output.cpu().tolist() == expected.tolist()