From 419fea3370467bc95675fc76891416375c9cff5e Mon Sep 17 00:00:00 2001
From: Madhur Tandon <madhurtandon23@gmail.com>
Date: Mon, 8 Aug 2022 19:42:23 +0530
Subject: [PATCH] remove get client from state

---
 mlem/contrib/kubernetes/base.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/mlem/contrib/kubernetes/base.py b/mlem/contrib/kubernetes/base.py
index 574193bf..cf604d2c 100644
--- a/mlem/contrib/kubernetes/base.py
+++ b/mlem/contrib/kubernetes/base.py
@@ -38,13 +38,6 @@ class K8sDeploymentState(DeployState):
     type: ClassVar = "kubernetes"
     image: Optional[DockerImage]
     pod_name: Optional[str]
-    kube_config_file_path: Optional[str] = None
-
-    def get_client(self):
-        config.load_kube_config(
-            config_file=self.kube_config_file_path
-            or os.getenv("KUBECONFIG", default="~/.kube/config")
-        )
 
 
 class K8sDeployment(MlemDeployment):
@@ -52,6 +45,13 @@ class K8sDeployment(MlemDeployment):
     server: Optional[Server] = None
     state_type: ClassVar = K8sDeploymentState
     image_name: Optional[str] = None
+    kube_config_file_path: Optional[str] = None
+
+    def _get_client(self, state: K8sDeploymentState):
+        config.load_kube_config(
+            config_file=self.kube_config_file_path
+            or os.getenv("KUBECONFIG", default="~/.kube/config")
+        )
 
 
 class K8sEnv(MlemEnv[K8sDeployment]):