Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Only publish cu101 and cu101mkl variants to PyPI
Browse files Browse the repository at this point in the history
  • Loading branch information
perdasilva committed Sep 30, 2019
1 parent abd4755 commit 8b8443a
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion cd/python/pypi/Jenkins_pipeline.groovy
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,12 @@
// NOTE:
// ci_utils and cd_utils are loaded by the originating Jenkins job, e.g. jenkins/Jenkinsfile_release_job

// Only post the following variants to PyPI.
// This is a temporary solution until we are confident with the packages generated by CI
// This should be removed in the not too distant future.
// We only skip the publish step so we can still QA the other variants.
pypi_releases = ["cu101", "cu101mkl"]

def get_pipeline(mxnet_variant) {
def node_type = mxnet_variant.startsWith('cu') ? NODE_LINUX_GPU : NODE_LINUX_CPU
return cd_utils.generic_pipeline(mxnet_variant, this, node_type)
Expand Down Expand Up @@ -61,7 +67,11 @@ def test(mxnet_variant) {
def push(mxnet_variant) {
ws("workspace/python_pypi/${mxnet_variant}/${env.BUILD_NUMBER}") {
// publish package to pypi
sh "./ci/docker/runtime_functions.sh cd_pypi_publish"
if (mxnet_variant in pypi_releases) {
sh "./ci/docker/runtime_functions.sh cd_pypi_publish"
} else {
echo "Temporarily skipping publishing PyPI package for '${mxnet_variant}'."
}
}
}

Expand Down

0 comments on commit 8b8443a

Please sign in to comment.