@@ -205,7 +205,7 @@ def serialize(self) -> dict[str, Any]:
205205 serialized ["default_args_override" ] = self .default_args_override
206206 if len (self .axis_label_override ) > 0 :
207207 serialized ["axis_label_override" ] = cast (
208- PlotLabelOverrideMap , self .axis_label_override
208+ " PlotLabelOverrideMap" , self .axis_label_override
209209 ).model_dump ()
210210 return serialized
211211
@@ -270,7 +270,7 @@ def update_axes_labels(self, axes: Axes | Iterable[Axes]) -> None:
270270 if isinstance (ax , Axes ):
271271 orig_x_label = ax .get_xlabel ()
272272 orig_y_label = ax .get_ylabel ()
273- axis_label_override = cast (PlotLabelOverrideMap , self .axis_label_override )
273+ axis_label_override = cast (" PlotLabelOverrideMap" , self .axis_label_override )
274274
275275 if (
276276 override_label := axis_label_override .find_axis_label (orig_x_label , "x" )
@@ -362,19 +362,21 @@ def merge(self, other: PlotConfig) -> PlotConfig: # noqa: C901
362362 updated : dict [str , PerFunctionPlotConfig ] = {}
363363 # Update general field
364364 for key in self .model_fields_set :
365- updated [key ] = cast (PerFunctionPlotConfig , getattr (self , key ))
365+ updated [key ] = cast (" PerFunctionPlotConfig" , getattr (self , key ))
366366 if key in other .model_fields_set :
367- updated [key ] = updated [key ].merge (cast (PerFunctionPlotConfig , getattr (other , key )))
367+ updated [key ] = updated [key ].merge (
368+ cast ("PerFunctionPlotConfig" , getattr (other , key ))
369+ )
368370 for key in other .model_fields_set :
369371 if key not in updated :
370372 updated [key ] = getattr (other , key )
371373 # Update model_extra
372374 if self .model_extra is not None :
373375 for key , value in self .model_extra .items ():
374- updated [key ] = cast (PerFunctionPlotConfig , value )
376+ updated [key ] = cast (" PerFunctionPlotConfig" , value )
375377 if other .model_extra is not None and key in other .model_extra :
376378 updated [key ] = updated [key ].merge (
377- cast (PerFunctionPlotConfig , other .model_extra [key ])
379+ cast (" PerFunctionPlotConfig" , other .model_extra [key ])
378380 )
379381 if other .model_extra is not None :
380382 for key , value in other .model_extra .items ():
@@ -525,8 +527,8 @@ def wrapper(*args: Param.args, **kwargs: Param.kwargs) -> RetType: # noqa: DOC
525527 function_config = CONFIG .plotting .get_function_config (func .__name__ )
526528 override_kwargs = function_config .find_override_kwargs (not_user_provided_kwargs )
527529 updated_kwargs = kwargs | override_kwargs
528- arg_axes = find_axes (getcallargs (func , * args , ** updated_kwargs ).values ())
529- return_values = func (* args , ** updated_kwargs )
530+ arg_axes = find_axes (getcallargs (func , * args , ** updated_kwargs ).values ()) # type: ignore[arg-type]
531+ return_values = func (* args , ** updated_kwargs ) # type: ignore[arg-type]
530532 function_config .update_axes_labels (arg_axes )
531533
532534 if isinstance (return_values , Iterable ):
0 commit comments