From ee50497db2a4754f4bc064710a27dc759fc98d79 Mon Sep 17 00:00:00 2001
From: Frank Lee <somerlee.9@gmail.com>
Date: Thu, 26 May 2022 11:42:50 +0800
Subject: [PATCH] [ci] fixed nightly build workflow (#1029)

---
 .github/workflows/release_bdist.yml   | 2 +-
 .github/workflows/release_nightly.yml | 9 ++++++---
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/release_bdist.yml b/.github/workflows/release_bdist.yml
index e9f1c70b6..2e5233d0a 100644
--- a/.github/workflows/release_bdist.yml
+++ b/.github/workflows/release_bdist.yml
@@ -61,7 +61,7 @@ jobs:
           git_ref: ${{ github.event.inputs.github_ref }}
       - name: Build bdist wheel
         run: |
-          pip install beautifulsoup4 requests
+          pip install beautifulsoup4 requests packaging
           python ./build_colossalai_wheel.py
       - name: 🚀 Deploy
         uses: garygrossgarten/github-action-scp@release
diff --git a/.github/workflows/release_nightly.yml b/.github/workflows/release_nightly.yml
index 3ad078628..9651209a0 100644
--- a/.github/workflows/release_nightly.yml
+++ b/.github/workflows/release_nightly.yml
@@ -70,12 +70,15 @@ jobs:
     steps:
       - name: executing remote ssh commands using password
         uses: appleboy/ssh-action@master
+        env:
+          BUILD_DIR: ${{ secrets.PRIVATE_PYPI_NIGHTLY_DIR }}
         with:
           host: ${{ secrets.PRIVATE_PYPI_HOST }}
           username: ${{ secrets.PRIVATE_PYPI_USER }}
           password: ${{ secrets.PRIVATE_PYPI_PASSWD }}
+          envs: BUILD_DIR
           script: |
-            cd $build_dir
+            cd $BUILD_DIR
             find . -type f -mtime +0 -exec rm -f {} +
-        env:
-          build_dir: ${{ secrets.PRIVATE_PYPI_NIGHTLY_DIR }}
+          script_stop: true
+