From 72d5f2971c45eca3a87f30fe982471297e551531 Mon Sep 17 00:00:00 2001 From: Ramana Radhakrishnan Date: Sat, 24 Oct 2020 08:27:30 +0100 Subject: [PATCH] Add pytest-xdist and pytest-profiling to the base installation packages. (#6736) For building and testing some small portions of the python testsuite, I've been playing off and on with xdist and pytest-profiling. We know it's not safe for the entirity of CI yet but this could enable smaller parts of pipelines that folks use using the common scripts to be parallelized or indeed profiled for more insight into where time is spent in building and testing TVM --- docker/install/ubuntu_install_python_package.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/install/ubuntu_install_python_package.sh b/docker/install/ubuntu_install_python_package.sh index 2ed14c2736789..c8d9856b6de08 100755 --- a/docker/install/ubuntu_install_python_package.sh +++ b/docker/install/ubuntu_install_python_package.sh @@ -21,4 +21,4 @@ set -u set -o pipefail # install libraries for python package on ubuntu -pip3 install six numpy pytest cython decorator scipy tornado typed_ast pytest mypy orderedset attrs requests Pillow packaging cloudpickle +pip3 install six numpy pytest cython decorator scipy tornado typed_ast pytest pytest-xdist pytest-profiling mypy orderedset attrs requests Pillow packaging cloudpickle