try to fix pre-commit action and update Docker def

.github/workflows/pre-commit.yml

@@ -17,4 +17,4 @@ jobs:
       with:
         path: ~/.cache/pre-commit
         key: pre-commit|${{ env.PY }}|${{ hashFiles('.pre-commit-config.yaml') }}
-    - uses: pre-commit/action@v1.0.1
+    - uses: pre-commit/action@v2.0.0

Dockerfile

@@ -1,11 +1,6 @@
 FROM mcr.microsoft.com/azureml/base-gpu:openmpi3.1.2-cuda10.1-cudnn7-ubuntu18.04
 
-RUN python3 -m pip install --upgrade --no-cache-dir numpy scipy docopt torch torchvision dpu-utils more-itertools typing_extensions sentencepiece azureml-sdk pyyaml dill jellyfish
-
+RUN python3 -m pip install --upgrade --no-cache-dir numpy scipy docopt dpu-utils more-itertools typing_extensions sentencepiece azureml-sdk pyyaml dill jellyfish
+RUN pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
 # Install torch scatter
-RUN export FORCE_CUDA=1
-RUN git clone https://github.com/rusty1s/pytorch_scatter.git
-RUN cd pytorch_scatter && git checkout '2.0.3' && FORCE_CUDA=1 python3 ./setup.py build && python3 ./setup.py install && cd .. && rm -rf ./pytorch_scatter
-
-# Test installation
-RUN python3 -c "import torch_scatter"
+RUN pip install torch-scatter==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html