remove get client from state

iterative · Aug 8, 2022 · 419fea3 · 419fea3
1 parent bccb4d8
commit 419fea3
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/mlem/contrib/kubernetes/base.py b/mlem/contrib/kubernetes/base.py
@@ -38,20 +38,20 @@ 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):
     type: ClassVar = "kubernetes"
     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]):