From 77049e7087988b96552aa313cecc92fbb2a068db Mon Sep 17 00:00:00 2001
From: Daniel Ciborowski <dciborow@microsoft.com>
Date: Tue, 26 Apr 2022 16:38:22 -0400
Subject: [PATCH] Update action.yml

---
 action.yml | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/action.yml b/action.yml
index 13d047d..4092481 100644
--- a/action.yml
+++ b/action.yml
@@ -142,17 +142,16 @@ runs:
     - uses: actions/setup-python@v2.2.2
       with:
         python-version: 3.7
-    - run: |
+    - if ${{ inputs.pypi_publish || inputs.pypi_custom_publish }}
+      run: |
         python -m pip install --upgrade pip
         python -m pip install flit
         
-        if ${{ inputs.pypi_publish }}; then
-          python -m flit publish
-        fi
-        
         if ${{ inputs.pypi_custom_publish }}; then
-          python -m flit publish --repository ${{ inputs.pypi_repo }}
+          repository="--repository ${{ inputs.pypi_repo }}"
         fi
+
+        python -m flit publish $repository
       env:
         FLIT_USERNAME: __token__
         FLIT_PASSWORD: ${{ inputs.pypi_password }}