Add test channels for pytorch version functions (#2208)

7aea80c9 · Eli Uriegas · GitHub · 40ad5572 · 7aea80c9
Unverified Commit 7aea80c9 authored May 12, 2020 by Eli Uriegas Committed by GitHub May 12, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

packaging/pkg_helpers.bash packaging/pkg_helpers.bash +2 -1

No files found.
--- a/packaging/pkg_helpers.bash
+++ b/packaging/pkg_helpers.bash
@@ -214,6 +214,7 @@ setup_pip_pytorch_version() {
  else
    pip_install "torch==$PYTORCH_VERSION$PYTORCH_VERSION_SUFFIX" \
      -f https://download.pytorch.org/whl/torch_stable.html \
+      -f https://download.pytorch.org/whl/test/torch_test.html
      -f https://download.pytorch.org/whl/nightly/torch_nightly.html
  fi
 }
@@ -239,7 +240,7 @@ setup_conda_pytorch_constraint() {
      exit 1
    fi
  else
-    export CONDA_CHANNEL_FLAGS="-c pytorch -c pytorch-nightly"
+    export CONDA_CHANNEL_FLAGS="-c pytorch -c pytorch-nightly -c pytorch-test"
  fi
  if [[ "$CU_VERSION" == cpu ]]; then
    export CONDA_PYTORCH_BUILD_CONSTRAINT="- pytorch==$PYTORCH_VERSION${PYTORCH_VERSION_SUFFIX}"