From 2d7efc46ddcd5465b797c9fe72dbcb1deab8035c Mon Sep 17 00:00:00 2001
From: agunapal <agunapal@ischool.berkeley.edu>
Date: Sun, 11 Feb 2024 18:34:48 +0000
Subject: [PATCH] add kserve nightly badge

---
 README.md                                     | 1 +
 kubernetes/kserve/tests/scripts/test_mnist.sh | 7 +++++++
 2 files changed, 8 insertions(+)

diff --git a/README.md b/README.md
index 5bf85eec92..a4aa612b07 100644
--- 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
index 9927b8a87d..cf4f2b4fb6 100755
--- 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