Fix

Signed-off-by: Tyler Gu <[email protected]>
xlab-uiuc · Nov 21, 2023 · f09922d · f09922d
1 parent 8d742ca
commit f09922d
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/performance_measurement/measure_performance.py b/performance_measurement/measure_performance.py
@@ -64,7 +64,6 @@ def test_normal(
     configuration.debug = False
     kubernetes.client.Configuration.set_default(configuration)
 
-
     # start the k8s cluster
     kubeconfig = os.path.join(os.path.expanduser("~"), ".kube", "anvil")
     deploy_metrics_server_f = partial(
@@ -103,8 +102,8 @@ def test_normal(
         watcher = WatchStats("anvil", control_plane_stats_dir)
         pods_watcher = MetricsApiWatcher(control_plane_stats_dir)
         watcher_thread = threading.Thread(target=watcher.start, args=(cluster,))
-        pods_watcher_thread = threading.Thread(
-            target=pods_watcher.start, args=(runner.apiclient, deploy.operator_name()))
+        pods_watcher_thread = threading.Thread(target=pods_watcher.start, args=(
+            kubernetes_client(kubeconfig, context_name), deploy.operator_name()))
         watcher_thread.start()
         pods_watcher_thread.start()
 
@@ -148,8 +147,8 @@ def test_normal(
                 logging.info('Not deployed. Try again!')
 
             watcher_thread = threading.Thread(target=watcher.start, args=(cluster,))
-            pods_watcher_thread = threading.Thread(
-                target=pods_watcher.start, args=(runner.apiclient, deploy.operator_name()))
+            pods_watcher_thread = threading.Thread( target=pods_watcher.start, args=(
+                kubernetes_client(kubeconfig, context_name), deploy.operator_name()))
             watcher_thread.start()
             pods_watcher_thread.start()