fix(test): overriding kfp version when loading component def from yaml (#12364)

Signed-off-by: Nelesh Singla <[email protected]>

Author: nsingla

sdk/python/kfp/compiler/compiler.py

@@ -52,6 +52,7 @@ def compile(
         pipeline_func: base_component.BaseComponent,
         package_path: str,
         pipeline_name: Optional[str] = None,
+        pipeline_display_name: Optional[str] = None,
         pipeline_parameters: Optional[Dict[str, Any]] = None,
         type_check: bool = True,
         kubernetes_manifest_options: Optional[
@@ -64,6 +65,7 @@ def compile(
             pipeline_func: Pipeline function constructed with the ``@dsl.pipeline`` or component constructed with the ``@dsl.component`` decorator.
             package_path: Output YAML file path. For example, ``'~/my_pipeline.yaml'`` or ``'~/my_component.yaml'``.
             pipeline_name: Name of the pipeline.
+            pipeline_display_name: Display name of the pipeline
             pipeline_parameters: Map of parameter names to argument values.
             type_check: Whether to enable type checking of component interfaces during compilation.
             kubernetes_manifest_options: KubernetesManifestOptions object for Kubernetes manifest output during pipeline compilation.
@@ -82,7 +84,7 @@ def compile(
                 pipeline_spec=pipeline_func.pipeline_spec,
                 pipeline_name=pipeline_name,
                 pipeline_parameters=pipeline_parameters,
-            )
+                pipeline_display_name=pipeline_display_name)
 
             builder.write_pipeline_spec_to_file(
                 pipeline_spec=pipeline_spec,

sdk/python/test/test_utils/comparison_utils.py

@@ -57,10 +57,14 @@ def compare_pipeline_spec_dicts(cls, actual: dict, expected: dict,
                     # Override SDK Version in the args as well to match the current version
                     if key == 'command':
                         for index, command in enumerate(value):
-                            if re.search("kfp==[0-9].[0-9]+.[0-9]+",
-                                         command) is not None:
-                                value[index] = re.sub(
-                                    "kfp==[0-9].[0-9]+.[0-9]+",
-                                    f"kfp=={kfp.__version__}", command)
+                            value[index] = cls.override_sdk_version(command)
+                        for index, command in enumerate(actual[key]):
+                            actual[key][index] = cls.override_sdk_version(
+                                command)
                     assert value == actual[
                         key], f'Value for "{key}" is not the same'
+
+    @classmethod
+    def override_sdk_version(cls, text: str) -> str:
+        return re.sub("kfp==[0-9].[0-9]+.[0-9]+", f"kfp=={kfp.__version__}",
+                      text)