Fixes KubernetesClusterConfig import in prefect-kubernetes (#14154)

PrefectHQ · Jun 19, 2024 · e42d112 · e42d112
1 parent 09a459f
commit e42d112
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/docs/integrations/prefect-kubernetes/index.mdx b/docs/integrations/prefect-kubernetes/index.mdx
@@ -71,8 +71,7 @@ if __name__ == "__main__":
 
 ```python
 # with minikube / docker desktop & a valid ~/.kube/config this should ~just work~
-from prefect.blocks.kubernetes import KubernetesClusterConfig
-from prefect_kubernetes.credentials import KubernetesCredentials
+from prefect_kubernetes.credentials import KubernetesCredentials, KubernetesClusterConfig
 
 k8s_config = KubernetesClusterConfig.from_file('~/.kube/config')
 

diff --git a/src/integrations/prefect-kubernetes/README.md b/src/integrations/prefect-kubernetes/README.md
@@ -81,8 +81,7 @@ if __name__ == "__main__":
 
 ```python
 # with minikube / docker desktop & a valid ~/.kube/config this should ~just work~
-from prefect.blocks.kubernetes import KubernetesClusterConfig
-from prefect_kubernetes.credentials import KubernetesCredentials
+from prefect_kubernetes.credentials import KubernetesCredentials, KubernetesClusterConfig
 
 k8s_config = KubernetesClusterConfig.from_file('~/.kube/config')
 

diff --git a/src/integrations/prefect-kubernetes/prefect_kubernetes/worker.py b/src/integrations/prefect-kubernetes/prefect_kubernetes/worker.py
@@ -121,7 +121,6 @@
 from typing_extensions import Literal, Self
 
 import prefect
-from prefect.blocks.kubernetes import KubernetesClusterConfig
 from prefect.exceptions import (
     InfrastructureError,
     InfrastructureNotAvailable,
@@ -140,6 +139,7 @@
     BaseWorker,
     BaseWorkerResult,
 )
+from prefect_kubernetes.credentials import KubernetesClusterConfig
 from prefect_kubernetes.events import KubernetesEventsReplicator
 from prefect_kubernetes.utilities import (
     _slugify_label_key,