add kserve nightly badge (#2939)

pytorch · Feb 11, 2024 · 33d87e3 · 33d87e3
1 parent e6654ec
commit 33d87e3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -4,6 +4,7 @@
 ![Docker Nightly build](https://github.com/pytorch/serve/actions/workflows/docker-nightly-build.yml/badge.svg)
 ![Benchmark Nightly](https://github.com/pytorch/serve/actions/workflows/benchmark_nightly.yml/badge.svg)
 ![Docker Regression Nightly](https://github.com/pytorch/serve/actions/workflows/regression_tests_docker.yml/badge.svg)
+![KServe Regression Nightly](https://github.com/pytorch/serve/actions/workflows/kserve_cpu_tests.yml/badge.svg)
 
 TorchServe is a flexible and easy-to-use tool for serving and scaling PyTorch models in production.
 

diff --git a/kubernetes/kserve/tests/scripts/test_mnist.sh b/kubernetes/kserve/tests/scripts/test_mnist.sh
@@ -45,6 +45,7 @@ function make_cluster_accessible() {
     EXPECTED="$4"
     if [ "${PREDICTION}" = "${EXPECTED}" ]; then
         echo "✓ SUCCESS"
+        cleanup_port_forwarding
     else
         echo "✘ Test failed: Prediction: ${PREDICTION}, expected ${EXPECTED}."
         delete_minikube_cluster
@@ -75,6 +76,7 @@ function make_cluster_accessible_for_grpc() {
     EXPECTED="$4"
     if [ "${PREDICTION}" = "${EXPECTED}" ]; then
         echo "✓ SUCCESS"
+        cleanup_port_forwarding
     else
         echo "✘ Test failed: Prediction: ${PREDICTION}, expected ${EXPECTED}."
         delete_minikube_cluster
@@ -157,6 +159,11 @@ function wait_for_port_forwarding() {
     sleep "$interval"
 }
 
+function cleanup_port_forwarding() {
+    echo "Clean up port forwarding"
+    pkill kubectl
+}
+
 export INGRESS_HOST=localhost
 export INGRESS_PORT=8080
 export MODEL_NAME=mnist