From 045d460327e6adc9be9929fd9537d42bc57e6223 Mon Sep 17 00:00:00 2001
From: Michael Mi <guocuimi@gmail.com>
Date: Mon, 14 Oct 2024 19:28:18 -0700
Subject: [PATCH] update

---
 .github/workflows/package_test.yml | 3 +--
 .gitignore                         | 3 +++
 scripts/run_pytest.sh              | 9 +++------
 3 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/package_test.yml b/.github/workflows/package_test.yml
index a0d5ff18..8c0280c3 100644
--- a/.github/workflows/package_test.yml
+++ b/.github/workflows/package_test.yml
@@ -87,7 +87,6 @@ jobs:
     - name: Install the package and run pytest
       timeout-minutes: 10
       run: |
-        docker pull pytorch/manylinux-builder:cuda${CUDA_VERSION}
         docker run --rm -t --gpus=all \
             -v "$CI_CACHE_DIR":/ci_cache \
             -v "$GITHUB_WORKSPACE":/ScaleLLM \
@@ -96,5 +95,5 @@ jobs:
             -e TORCH_VERSION=${TORCH_VERSION} \
             -e PIP_CACHE_DIR=/ci_cache/.pip \
             -u $(id -u):$(id -g) \
-            pytorch/manylinux-builder:cuda${CUDA_VERSION} \
+            vectorchai/scalellm_manylinux:cuda${CUDA_VERSION} \
             bash /ScaleLLM/scripts/run_pytest.sh
diff --git a/.gitignore b/.gitignore
index d7cd9f22..481e9b7a 100644
--- a/.gitignore
+++ b/.gitignore
@@ -13,6 +13,9 @@
 # deps
 /.deps
 
+# pip cache
+/.pip
+
 # libtorch
 /libtorch
 
diff --git a/scripts/run_pytest.sh b/scripts/run_pytest.sh
index 8011b202..4a47af8f 100755
--- a/scripts/run_pytest.sh
+++ b/scripts/run_pytest.sh
@@ -14,6 +14,7 @@ ensure_env TORCH_VERSION
 ensure_env CUDA_VERSION
 
 PROJECT_ROOT="$(cd "$(dirname "$0")/.." && pwd)"
+cd "$PROJECT_ROOT"
 
 export HOME=/tmp/home
 mkdir -p $HOME
@@ -26,16 +27,12 @@ export PATH="/opt/python/cp${PYVER}-cp${PYVER}/bin:$PATH"
 # install PyTorch
 pip install torch==$TORCH_VERSION -i "https://download.pytorch.org/whl/cu${CUDA_VERSION//./}"
 
-cd "$PROJECT_ROOT"
-
 # install dependencies
-pip install numpy
 pip install -r requirements-test.txt
 
 # install scalellm wheel
 pip install dist/*.whl
 
 # run pytest
-printf "\n\nRunning pytest\n\n"
-# cd tests
-python3 -m pytest
\ No newline at end of file
+cd tests
+pytest
\ No newline at end of file