Add variable to make it possible to test weekly onnx package (#1627)

Signed-off-by: Tom Wildenhain <[email protected]>

Author: TomWildenhain-Microsoft

ci_build/azure_pipelines/templates/setup.yml

@@ -12,11 +12,6 @@ steps:
     then
         pip install $(CI_PIP_ONNX_NAME) $(CI_PIP_ONNX_BACKEND_NAME) numpy --no-deps -U
     fi
-    if [[ $ONNX_TEST ==  "true" ]] ;
-    then
-      pip uninstall -y onnx
-      pip install --index-url https://test.pypi.org/simple/ onnx
-    fi
     if [[ $CI_ONNXRUNTIME_NIGHTLY == "true" ]] ;
     then
       pip uninstall -y onnxruntime
@@ -49,6 +44,14 @@ steps:
     fi
 
     python setup.py install
+
+    if [[ $ONNX_TEST ==  "true" ]] ;
+    then
+      # onnx-weekly won't satisfy onnx requirement so uninstallation must happen here
+      pip uninstall -y onnx
+      pip install --index-url https://test.pypi.org/simple/ ${PIP_ONNX_NAME:-onnx}
+    fi
+
     pip freeze --all
   displayName: 'Setup Environment'