@@ -195,15 +195,15 @@ def build(
195195 if outputs is None :
196196 outputs = {}
197197
198- jobs = self ._update_nodes_variable_names (_locals )
198+ jobs : Dict = self ._update_nodes_variable_names (_locals )
199199 pipeline_component = PipelineComponent (
200200 name = self .name ,
201201 version = self .version ,
202202 display_name = self .display_name ,
203203 description = self .description ,
204204 inputs = self .inputs ,
205205 jobs = jobs ,
206- tags = self .tags ,
206+ tags = self .tags , # type: ignore[arg-type]
207207 source_path = self .source_path ,
208208 _source = ComponentSource .DSL ,
209209 )
@@ -276,10 +276,14 @@ def _map_internal_output_type(_meta: Output) -> str:
276276 """
277277 if type (_meta ).__name__ != "InternalOutput" :
278278 return str (_meta .type )
279- return str (_meta .map_pipeline_output_type ())
279+ return str (_meta .map_pipeline_output_type ()) # type: ignore[attr-defined]
280280
281281 # Note: Here we set PipelineOutput as Pipeline's output definition as we need output binding.
282- output_meta = Output (type = _map_internal_output_type (meta ), description = meta .description , mode = meta .mode )
282+ output_meta = Output (
283+ type = _map_internal_output_type (meta ), # type: ignore[arg-type]
284+ description = meta .description ,
285+ mode = meta .mode ,
286+ )
283287 pipeline_output = PipelineOutput (
284288 port_name = key ,
285289 data = None ,
@@ -292,9 +296,11 @@ def _map_internal_output_type(_meta: Output) -> str:
292296 binding_output = value ,
293297 )
294298 # copy node level output setting to pipeline output
295- copy_output_setting (source = value ._owner .outputs [value ._port_name ], target = pipeline_output )
299+ copy_output_setting (
300+ source = value ._owner .outputs [value ._port_name ], target = pipeline_output # type: ignore[arg-type]
301+ )
296302
297- value ._owner .outputs [value ._port_name ]._data = pipeline_output
303+ value ._owner .outputs [value ._port_name ]._data = pipeline_output # type: ignore[union-attr]
298304
299305 output_dict [key ] = pipeline_output
300306 output_meta_dict [key ] = output_meta ._to_dict ()
@@ -383,7 +389,7 @@ def _get_name_or_component_name(node: Union[BaseNode, AutoMLJob]) -> Optional[Un
383389 f"node name: { name !r}
384390 )
385391 local_names .add (name )
386- id_name_dict [v ._instance_id ] = name
392+ id_name_dict [v ._instance_id ] = name # type: ignore[union-attr]
387393 name_count_dict [name ] = 1
388394
389395 # Find the last user-defined name for the same type of components
@@ -420,6 +426,7 @@ def _update_inputs(self, pipeline_inputs: Dict[str, Union[PipelineInput, Input,
420426 :type pipeline_inputs: Dict[str, Union[PipelineInput, Input, NodeOutput, Any]]
421427 """
422428 for input_name , value in pipeline_inputs .items ():
429+ anno : Any = None
423430 if input_name not in self .inputs :
424431 if isinstance (value , PipelineInput ):
425432 value = value ._data
@@ -497,12 +504,15 @@ def _validate_inferred_outputs(self, output_meta_dict: dict, output_dict: Dict[s
497504 unmatched_outputs .append (
498505 f"{ key } { actual_output } { expected_output }
499506 )
507+ res = output_dict [key ]._meta
500508 if expected_description :
501- output_dict [key ]._meta .description = expected_description
509+ if res is not None :
510+ res .description = expected_description
502511 # also copy the description to pipeline job
503512 output_dict [key ].description = expected_description
504513 if expected_mode :
505- output_dict [key ]._meta .mode = expected_mode
514+ if res is not None :
515+ res .mode = expected_mode
506516 # also copy the mode to pipeline job
507517 output_dict [key ].mode = expected_mode
508518
@@ -512,7 +522,7 @@ def _validate_inferred_outputs(self, output_meta_dict: dict, output_dict: Dict[s
512522 @staticmethod
513523 def _validate_keyword_in_node_io (node : Union [BaseNode , AutoMLJob ]) -> None :
514524 if has_attr_safe (node , "inputs" ):
515- for input_name in set (node .inputs ) & COMPONENT_IO_KEYWORDS :
525+ for input_name in set (node .inputs ) & COMPONENT_IO_KEYWORDS : # type: ignore[arg-type]
516526 module_logger .warning (
517527 'Reserved word "%s" is used as input name in node "%s", '
518528 "can only be accessed with '%s.inputs[\" %s\" ]'" ,
@@ -522,7 +532,7 @@ def _validate_keyword_in_node_io(node: Union[BaseNode, AutoMLJob]) -> None:
522532 input_name ,
523533 )
524534 if has_attr_safe (node , "outputs" ):
525- for output_name in set (node .outputs ) & COMPONENT_IO_KEYWORDS :
535+ for output_name in set (node .outputs ) & COMPONENT_IO_KEYWORDS : # type: ignore[arg-type]
526536 module_logger .warning (
527537 'Reserved word "%s" is used as output name in node "%s", '
528538 "can only be accessed with '%s.outputs[\" %s\" ]'" ,
0 commit comments