fix formatting

kubeflow · Jun 10, 2024 · 959fc6c · 959fc6c
1 parent 1b5123b
commit 959fc6c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/sdk/python/kfp/compiler/compiler_utils.py b/sdk/python/kfp/compiler/compiler_utils.py
@@ -15,7 +15,7 @@
 
 import collections
 import copy
-from typing import Any, DefaultDict, Dict, List, Mapping, Set, Tuple, Union
+from typing import DefaultDict, Dict, List, Mapping, Set, Tuple, Union
 
 from kfp import dsl
 from kfp.dsl import for_loop
@@ -775,8 +775,8 @@ def get_dependencies(
 
 
 def recursive_replace(
-        data: Union[Dict, List[Any]], old_value: Union[int, float, bool, str],
-        new_value: Union[int, float, bool, str]) -> Union[Dict, List[Any]]:
+        data: Union[Dict, List], old_value: Union[int, float, bool, str],
+        new_value: Union[int, float, bool, str]) -> Union[Dict, List]:
     """Replaces values in a nested dict/list object.
 
     Args:

diff --git a/sdk/python/kfp/compiler/pipeline_spec_builder.py b/sdk/python/kfp/compiler/pipeline_spec_builder.py
@@ -49,12 +49,12 @@
 
 
 def replace_and_inject_placeholders(
-    input_value: Union[str, Dict, List[Any]],
+    input_value: Union[str, Dict, List],
     pipeline_task_spec: pipeline_spec_pb2.PipelineTaskSpec,
     task: pipeline_task.PipelineTask,
     parent_component_inputs: pipeline_spec_pb2.ComponentInputsSpec,
     tasks_in_current_dag: List[str]
-) -> tuple[Union[str, Dict, List[Any]], pipeline_spec_pb2.PipelineTaskSpec]:
+) -> Tuple[Union[str, Dict, List], pipeline_spec_pb2.PipelineTaskSpec]:
     """Replaces placeholders in task input and injects placeholders as
     component inputs.