Add import

Signed-off-by: Jason Parraga <sovietaced@gmail.com>
flyteorg · Feb 6, 2025 · 339a1bc · 339a1bc
1 parent 1d45a20
commit 339a1bc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/ray_plugin/ray_plugin/ray_example.py b/examples/ray_plugin/ray_plugin/ray_example.py
@@ -31,6 +31,7 @@
 )
 
 import ray
+from flytekit.models.task import K8sPod
 from flytekitplugins.ray import HeadNodeConfig, RayJobConfig, WorkerNodeConfig