@@ -621,7 +621,7 @@ def update_custom_metadata(
621621 self ._api .put_run_custom_metadata_v1_runs_run_id_custom_metadata_put (
622622 self .run_id ,
623623 custom_metadata_update_request = CustomMetadataUpdateRequest (
624- custom_metadata = cast (" dict[str, Any]" , convert_to_json_serializable (custom_metadata )),
624+ custom_metadata = cast (dict [str , Any ], convert_to_json_serializable (custom_metadata )), # noqa: TC006
625625 custom_metadata_checksum = custom_metadata_checksum ,
626626 ),
627627 _request_timeout = settings ().run_submit_timeout ,
@@ -660,7 +660,7 @@ def update_item_custom_metadata(
660660 self .run_id ,
661661 external_id ,
662662 custom_metadata_update_request = CustomMetadataUpdateRequest (
663- custom_metadata = cast (" dict[str, Any]" , convert_to_json_serializable (custom_metadata )),
663+ custom_metadata = cast (dict [str , Any ], convert_to_json_serializable (custom_metadata )), # noqa: TC006
664664 custom_metadata_checksum = custom_metadata_checksum ,
665665 ),
666666 _request_timeout = settings ().run_submit_timeout ,
@@ -761,7 +761,7 @@ def submit(
761761 payload = RunCreationRequest (
762762 application_id = application_id ,
763763 version_number = application_version ,
764- custom_metadata = cast (" dict[str, Any]" , convert_to_json_serializable (custom_metadata )),
764+ custom_metadata = cast (dict [str , Any ], convert_to_json_serializable (custom_metadata )), # noqa: TC006
765765 items = items ,
766766 scheduling = scheduling ,
767767 )
@@ -940,7 +940,7 @@ def _amend_input_items_with_sdk_metadata(items: builtins.list[ItemCreationReques
940940 validate_item_sdk_metadata (base_sdk_metadata )
941941 item_custom_metadata ["sdk" ] = base_sdk_metadata
942942
943- item .custom_metadata = cast (" dict[str, Any]" , convert_to_json_serializable (item_custom_metadata ))
943+ item .custom_metadata = cast (dict [str , Any ], convert_to_json_serializable (item_custom_metadata )) # noqa: TC006
944944
945945 def _validate_input_items (self , payload : RunCreationRequest ) -> None :
946946 """Validates the input items in a run creation request.
0 commit comments