Minor change.

Signed-off-by: Revital Sur <[email protected]>
IBM · Sep 25, 2024 · 35b9989 · 35b9989
1 parent 4b6e8d2
commit 35b9989
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/kfp/kfp_support_lib/shared_workflow_support/src/runtime_utils/remote_jobs_utils.py b/kfp/kfp_support_lib/shared_workflow_support/src/runtime_utils/remote_jobs_utils.py
@@ -25,7 +25,6 @@
     ClusterSpec,
     HeadNodeSpec,
     RayJobRequest,
-    Template,
     template_decoder,
     WorkerNodeSpec,
     environment_variables_decoder,
@@ -122,7 +121,7 @@ def create_ray_cluster(
         """
         # start with templates
         # head_node
-        dct= {}
+        dct = {}
         dct["cpu"] = head_node.get("cpu", 1)
         dct["memory"] = head_node.get("memory", 1)
         dct["gpu"] = head_node.get("gpu", 0)
@@ -141,7 +140,7 @@ def create_ray_cluster(
         index = 0
         # For every worker group
         for worker_node in worker_nodes:
-            dct={}
+            dct ={}
             dct["cpu"] = worker_node.get("cpu", 1)
             dct["memory"] = worker_node.get("memory", 1)
             dct["gpu"] = worker_node.get("gpu", 0)