chore(e2e): install the correct version in e2e when run job (#2553)

star-whale · Jul 25, 2023 · 2bd0b74 · 2bd0b74
1 parent a7ede5b
commit 2bd0b74
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 3 deletions.
diff --git a/docker/external/base-entrypoint.sh b/docker/external/base-entrypoint.sh
@@ -18,7 +18,7 @@ welcome() {
     echo "StarWhale Base Entrypoint"
     echo "Date: `date -u +%Y-%m-%dT%H:%M:%SZ`"
     echo "Starwhale Version: ${SW_VERSION}"
-    echo "Python Version: ${SW_RUNTIME_VERSION}"
+    echo "Python Version: ${PYTHON_VERSION}"
     echo "Runtime Restored: ${RUNTIME_RESTORED}"
     echo "Command type(Whether use custom command): ${USE_CUSTOM_CMD}"
     echo "Run: $1"

diff --git a/example/runtime/pytorch-e2e/requirements.txt b/example/runtime/pytorch-e2e/requirements.txt
@@ -1,4 +1,5 @@
-starwhale[serve]
+# use e2e version
+starwhale[serve]==${PYPI_RELEASE_VERSION}
 # for pfp
 pycocotools
 # for ucf101
@@ -16,4 +17,4 @@ torch==1.13.1; python_version=='3.7'
 torchaudio
 torchdata
 torchtext
-torchvision
+torchvision
diff --git a/scripts/client_test/update_controller_setting.sh b/scripts/client_test/update_controller_setting.sh
@@ -53,6 +53,17 @@ resourcePoolSetting:
     - name: "memory"
       defaults: 3145728
     - name: "nvidia.com/gpu"
+pypiSetting:
+  indexUrl: "http://$NEXUS_HOSTNAME:$PORT_NEXUS/repository/$REPO_NAME_PYPI/simple"
+  extraIndexUrl: "$SW_PYPI_EXTRA_INDEX_URL"
+  trustedHost: "$NEXUS_HOSTNAME"
+  retries: 10
+  timeout: 90
+datasetBuild:
+  resourcePool: "default"
+  image: "docker-registry.starwhale.cn/star-whale/base:latest"
+  clientVersion: "$PYPI_RELEASE_VERSION"
+  pythonVersion: "3.10"
 EOF
   )
   curl -X 'POST' \