| #!/usr/bin/env bash |
| |
| set -eou pipefail |
| |
| DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )" |
| source "${DIR}/common_utils.sh" |
| |
| # Allow for users to pass PACKAGE_NAME |
| # For use with other packages, i.e. torchvision, etc. |
| PACKAGE_NAME=${PACKAGE_NAME:-torch} |
| PACKAGE_TYPE=${PACKAGE_TYPE:-whl} |
| |
| PYTORCH_S3_BUCKET=${PYTORCH_S3_BUCKET:-s3://pytorch} |
| FROM=${FROM:-test} |
| PYTORCH_S3_FROM=${PYTORCH_S3_FROM:-${PYTORCH_S3_BUCKET}/${PACKAGE_TYPE}/${FROM}} |
| TO=${TO:-} |
| PYTORCH_S3_TO=${PYTORCH_S3_TO:-${PYTORCH_S3_BUCKET}/${PACKAGE_TYPE}/${TO}} |
| |
| aws_promote "${PACKAGE_NAME}" |