[Release 0.3.0] Fix PYPI upload

pytorch · Mar 10, 2022 · fbf097d · fbf097d
1 parent 38b2ac9
commit fbf097d
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/.github/workflows/_build_test_upload.yml b/.github/workflows/_build_test_upload.yml
@@ -216,7 +216,7 @@ jobs:
  PYPI_TOKEN: ${{ secrets.PYPI_TOKEN }}
  run: |
  pip3 install twine
- python -m twine upload
+ python -m twine upload \
  --username __token__ \
  --password "$PYPI_TOKEN" \
  dist/torchdata*.whl
@@ -266,12 +266,15 @@ jobs:
  else
  PYTORCH_VERSION=torch
  fi
- # Use test channel for official release
+
  PIP_CHANNEL=${{ needs.get_release_type.outputs.type }}
+
  if [[ $PIP_CHANNEL == 'official' ]]; then
- PIP_CHANNEL='test'
+ pip3 install "$PYTORCH_VERSION"
+ else
+ pip3 install --pre "$PYTORCH_VERSION" -f "https://download.pytorch.org/whl/$PIP_CHANNEL/cpu/torch_$PIP_CHANNEL.html"
  fi
- pip3 install --pre "$PYTORCH_VERSION" -f "https://download.pytorch.org/whl/$PIP_CHANNEL/cpu/torch_$PIP_CHANNEL.html"
+
  if [[ ${{ needs.get_release_type.outputs.type }} == 'nightly' ]]; then
  python setup.py install --nightly
  else