From be693cd094bcd20f9ec8fad0b7023589d4de9534 Mon Sep 17 00:00:00 2001
From: Yaning Cui <39021445+emotionor@users.noreply.github.com>
Date: Mon, 26 May 2025 11:33:47 +0800
Subject: [PATCH] [chore] Update docker.yml.

---
 .github/workflows/docker.yml | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml
index 6deaf4a..010744a 100644
--- a/.github/workflows/docker.yml
+++ b/.github/workflows/docker.yml
@@ -4,6 +4,7 @@ on:
   push:
     branches:
     - main
+    - pypi_test
 
 jobs:
   publish_package:
@@ -22,7 +23,6 @@ jobs:
           FLASH_ATTENTION_SKIP_CUDA_BUILD: "TRUE"
         run: |
           pip install setuptools wheel twine
-          cd unimol_tools
           python setup.py sdist --dist-dir=dist
 
       - name: Deploy
@@ -30,5 +30,4 @@ jobs:
           TWINE_USERNAME: "__token__"
           TWINE_PASSWORD: ${{ secrets.PYPI_TOKEN }}
         run: |
-          cd unimol_tools
-          python -m twine upload dist/*
\ No newline at end of file
+          python -m twine upload dist/*