Address review comment.

Signed-off-by: Revital Sur <[email protected]>

kfp/doc/simple_transform_pipeline.md

@@ -220,6 +220,7 @@ export KFP_TOLERATIONS='[{"key": "key","operator": "Equal", "value1": "value", "
 export KFP_NODE_SELECTOR='{"label_key":"cloud.google.com/gke-accelerator","label_value":"nvidia-tesla-p4"}'
 
 ```
+In KFP v1, setting `KFP_TOLERATIONS` will apply to the Ray pods, overriding any tolerations specified in the `ray_head_options` and `ray_worker_options` pipeline parameters if they are present.
 
 ## Compiling a pipeline <a name = "compilation"></a>
 

kfp/kfp_support_lib/kfp_v1_workflow_support/src/workflow_support/compile_utils/component.py

@@ -62,7 +62,7 @@ def _add_tolerations() -> None:
                 if tolerations != "":
                     print(f"Note: Applying Tolerations {tolerations} to kfp and ray pods")
 
-                    # Add Tolerations as env var so it can used when creating the ray cluster
+                    # Add Tolerations as env var so it can be used when creating the ray cluster
                     component.add_env_variable(k8s_client.V1EnvVar(name="KFP_TOLERATIONS", value=tolerations))
 
                     tolerations = json.loads(tolerations)

kfp/kfp_support_lib/kfp_v2_workflow_support/src/workflow_support/compile_utils/component.py

@@ -42,13 +42,9 @@ def _add_tolerations() -> None:
             try:
                 tolerations = os.getenv("KFP_TOLERATIONS", "")
                 if tolerations != "":
-                    print(f"Note: Applying Tolerations {tolerations} to kfp and ray pods")
-
-                    # Add Tolerations as env var so it can used when creating the ray cluster
-                    task.add_pod_annotation("ray_tolerations", tolerations)
-                    kubernetes.use_field_path_as_env(
-                        task, env_name="KFP_TOLERATIONS", field_path="metadata.annotations['ray_tolerations']"
-                    )
+                    # TODO: apply the tolerations defined as env vars to ray pods.
+                    # Currently they can be specified in the pipeline params:
+                    # ray_head_options and ray_worker_options.
 
                     tolerations = json.loads(tolerations)
                     for toleration in tolerations: