Update pytest version used on CI (#19491)

* Update pytest version

* Update Dockerfile.test.arm

* Update python.sh

ci/docker/Dockerfile.test.arm

@@ -35,11 +35,11 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y \
 # Python dependencies
 RUN pip3 install --no-cache-dir --upgrade pip && \
     pip3 install --no-cache-dir \
-    pytest==5.3.5 \
+    pytest==6.1.2 \
     pytest-env==0.6.2 \
-    pytest-cov==2.8.1 \
-    pytest-xdist==1.31.0 \
-    pytest-timeout==1.3.4 \
+    pytest-cov==2.10.1 \
+    pytest-xdist==2.1.0 \
+    pytest-timeout==1.4.2 \
     mock==2.0.0
 
 ARG USER_ID=0

ci/docker/install/requirements

@@ -36,15 +36,14 @@ Cython==0.29.7
 cpplint==1.3.0
 pylint==2.3.1  # pylint and astroid need to be aligned
 astroid==2.3.3  # pylint and astroid need to be aligned
-pytest==5.3.5
+pytest==6.1.2
 pytest-env==0.6.2
-pytest-cov==2.8.1
-pytest-xdist==1.31.0
-pytest-timeout==1.3.4
-flaky==3.6.1
+pytest-cov==2.10.1
+pytest-xdist==2.1.0
+pytest-timeout==1.4.2
+flaky==3.7.0
 setuptools==49.6.0  # https://github.com/pypa/setuptools/issues/2352
 wheel
-mock==2.0.0
 
 # TVM dependencies
 decorator==4.4.0

docker/install/python.sh

@@ -24,4 +24,4 @@ apt-get update && apt-get install -y python3-dev
 # the version of the pip shipped with ubuntu may be too lower, install a recent version here
 cd /tmp && wget https://bootstrap.pypa.io/get-pip.py && python3 get-pip.py
 
-pip3 install pylint numpy requests Pillow pytest==5.3.2 pytest-env==0.6.2 pytest-cov==2.8.1 pytest-xdist==1.31.0
+pip3 install pylint numpy requests Pillow pytest==6.1.2 pytest-env==0.6.2 pytest-cov==2.10.1 pytest-xdist==2.1.0